1、安装背景
需要使用Python做一些深度学习上的应用,已安装Python、Anaconda、2017版VS、2019版VS
(我要安装的CUDA版本是12版本往后的版本,部分操作可能与12版本之前的安装操作不同)
2、CUDA的安装
首先需要确定自己电脑能安装哪个版本的CUDA
然后根据版本去CUDA官网点击对应的下载链接
这里一般选择本地下载(local),下载好后,运行exe,具体步骤可看下面这篇文章的第二部分:
3、CuDNN的下载
CuDNN版本要与CUDA版本对应,这里我CUDA-12.4版本选用的是CuDNN-9.1.0版本,CUDA-12.3版本选用的是CuDNN-8.9.7版本
以下是CuDNN的官方下载地址:
4、CuDNN安装以及相关环境配置
可参考以下文章进行CuDNN安装以及环境的配置:
5、Pytorch安装
Pytorch的版本要注意与CUDA版本对应,可以去官网复制粘贴安装命令,并在Anaconda中的目标虚拟环境中运行:
在Pytorch官网里面有让你选择Compute Platform的选项,一般只有CUDA12.1和12.4的选项,实际上,如果你的CUDA是12.3版本的,那么你可以选择Compute Platform为12.1
6、安装问题(待补充)
问题一:因为Nsight Compute安装失败而导致的CUDA安装失败
这个问题我在CUDA12.4版本安装中遇到过,解决办法就是单独下载这个模块,然后再运行CUDA的安装程序,在自定义安装中把把模块取消掉(因为已经下载过了),具体操作步骤可看下面这篇文章:
7、卸载
有的时候会因为版本冲突的原因,需要对CUDA或者Pytorch更换版本,下面简单介绍如何对它们进行卸载重装。
CUDA卸载及重装
在“控制面板中”点击“卸载程序”,然后搜素“cuda”,将相关程序卸载即可
重装的时候在CUDA的安装运行程序中,在自定义安装里面,只需勾选“CUDA”即可
Pytorch卸载
在Anaconda的目标虚拟环境中运行
pip uninstall torch