CUDA版本以及GPU驱动的对应关系_torch1.10.0对应的cuda

NVIDIA官网，看下CUDA版本以及GPU驱动的对应关系
安装GPU版本
CUDA11.3以及python3.8为例，我们打开网址：

https://download.pytorch.org/whl/cu113/torch_stable.html
1

检索下torch-1.10.0然后下载下面用红色框框住的文件：


接着检索下torchvision-0.11.1然后下载下面用红色框框住的文件

我们在这里可以找到我们需要的torch-1.10.0+cu113-cp38-cp38-win_amd64.whl以及torchvision-0.11.1+cu113-cp38-cp38-win_amd64.whl两个文件即可。注意，cu113代表CUDA11.3，cp38表示python3.8的编译环境，win_amd64表示WIN 64位操作系统。下载完成后

pip install torch-1.10.0+cu113-cp38-cp38-linux_x86_64.whl

1
2

pip install torchvision-0.11.1+cu113-cp38-cp38-linux_x86_64.whl

1
2