查看 CUDA 驱动版本
查看当前安装的 CUDA 版本,当前 CUDA 版本为 10.1
nvcc -V
![](https://i-blog.csdnimg.cn/blog_migrate/f4f5f28d15ee9fd613b0ac76559db9e9.png)
查看显卡驱动,以及支持的 CUDA 的最高版本,这里的12.0 指的是可驱动的最高版本。下载 CUDA 的版本应低于12.0
nvidia-smi
![](https://i-blog.csdnimg.cn/blog_migrate/81ec424ddd3c8a1013b95a3171698a21.png)
卸载旧版 CUDA
控制面板中卸载 CUDA
安装 CUDA
在 CUDA 官网 选择符合要求的版本安装
由于我安装的 pytorch 对应的 CUDA 版本是 11.7,所以我选择 11.7.1 版本
import torch
torch.cuda.is_available() # 查看pytorch是否支持CUDA
torch.cuda.device_count() # 查看可用的CUDA数量
torch.version.cuda # 查看对应CUDA的版本号
![](https://i-blog.csdnimg.cn/blog_migrate/5ab000bd49f7144fadf91b1e5691b366.png)
![](https://i-blog.csdnimg.cn/blog_migrate/67bfcef36eece0abdbbe3917baef4e94.png)
根据自己的系统选择并下载安装包
![](https://i-blog.csdnimg.cn/blog_migrate/05caa21f23ae47e12e83108ad5842d9a.png)
双击安装包,开始安装
![](https://i-blog.csdnimg.cn/blog_migrate/66aa65e94a272b87eb9c1d52e46868c2.png)
选择自定义安装
![](https://i-blog.csdnimg.cn/blog_migrate/b910641a27f6dd3a7e1085bbd5004a84.png)
我只卸载了CUDA,勾选CUDA
![](https://i-blog.csdnimg.cn/blog_migrate/bbf4fd57750f0a23ba6af80d1ea09d5f.png)
之后按默认设置,完成安装
![](https://i-blog.csdnimg.cn/blog_migrate/9f8fea86077d0f75493acf1057f52d7d.png)
安装完成
![](https://i-blog.csdnimg.cn/blog_migrate/f10cebcebf710dcd784e1c5931e90992.png)