三种方法，任选其一，推荐方法1、2
方法1、CUDA，pip/conda pytorch
CUDA
先去NVIDIA官网找CUDA，安装CUDA，再去pytorch官网找pytorch，安装pytorch
最新版CUDA https://developer.nvidia.com/cuda-downloads
历史版本CUDA https://developer.nvidia.com/cuda-toolkit-archive
根据需要选择
pip/conda pytorch
pytorch https://pytorch.org/get-started/locally/
历史版本pytorch https://pytorch.org/get-started/previous-versions/
根据需要选择
方法3、NVIDIA Driver，pip/conda pytorch
NVIDIA Driver
Windows驱动 https://www.nvidia.cn/Download/index.aspx?lang=cn
ubuntu 驱动
sudo apt search nvidia-driver | grep nvidia-driver
找一个合适的且较新的即可，还是根据需求选择
pip/conda pytorch
pytorch https://pytorch.org/get-started/locally/
历史版本pytorch https://pytorch.org/get-started/previous-versions/
根据需要选择
方法3、NVIDIA Driver，NVIDIA Docker
NVIDIA Driver
Windows驱动 https://www.nvidia.cn/Download/index.aspx?lang=cn
Linux 驱动 ，以Ubuntu举例
# 命令行输入
sudo apt search nvidia-driver | grep nvidia-driver
找一个合适的且较新的即可，还是根据需求选择