目录

一、Torch和对应cuda安装

1、查看本身电脑的cuda版本

2、查找对应cuda——torch——python版本

3、安装cuda

4、安装torch

二、window 10 NVIDIA cuda版本更新

---

一、Torch和对应cuda安装

项目使用torch想要使用GPU运行，但是报错，记录一下解决过程

！！！记住，最好重新建立新的虚拟环境，先安装cuda 再安装torch

conda create -n ccf python=3.7

以免之前的环境其他安装包有副作用影响

1、查看本身电脑的cuda版本

打开NVIDIA控制面板——帮助——系统信息——组件——查看cuda版本

2、查找对应cuda——torch——python版本

也可以通过在anaconda环境下输入命令查看可以安装的cuda版本和对应的安装限制

conda search cudatoolkit --info

3、安装cuda

在项目运行的虚拟环境下输入以下命令

因为第一步我的cuda版本是11.4

那可以安装10.2的，只要选择低于或者等于自身电脑cuda版本的就可以

conda install cudatoolkit=10.2

4、安装torch

这里也试过很多网上教程，有很多版本对应，下载完还是没有正常使用GPU，然后使用下面这个可以

这些是可以选择的torch版本，+cu110代表对应cuda版本是11.0

我之前安装了cuda版本是10.2，所以torch就要对应cu102

pip install torch===1.9.0+cu102 -f https://download.pytorch.org/whl/torch_stable.html

到这就可以正常运行啦

二、window 10 NVIDIA cuda版本更新

我一开始的电脑cuda版本很低，才9.0 就算按照上述方法下载torch版本是1.1.0 但是旧的torch版本很多功能都没有，最好还是提高电脑cuda版本

Please update your GPU driver by downloading and installing a new version 
from the URL: http://www.nvidia.com/Download/index.aspx 
Alternatively, go to:  https://pytorch.org 
to install a PyTorch version that has been compiled with your version

Official Drivers | NVIDIA

所以去到这个网址，选择自己系统的信息，找最新的cuda版本

来到这个页面，你如果填好信息，搜索，那给的是最新的版本，我下载后，不适合本机，可能版本调太高了，所以可以通过点击下面蓝笔圈出的beta and old drivers进入新页面

这里对应信息要填写好，可以通过任务管理器查看GPU版本

然后就下载保存运行，我下载的是

下载运行后，可以再次查看cuda版本，就可以了，成功从9.0升级到11.4

torch版本的选择可以更多，我就用10.2cuda     torch=1.9.0+cu102  大功告成