PS:python和pycharm我就不说啦,我的python是3.8的版本,pycharm是2020.3.5的版本
1 cuda首先鼠标右键,打开NVIDIA控制面板
然后点击右下角的系统信息查看能驱动的cuda版本
之后进入CUDA Toolkit Archive | NVIDIA Developer
选择相应的版本下载
我下载的是11.1的版本
直接往下安装,建议不要更改默认的路径,不然后面弄起来很麻烦
下载完成后
下一步就是为CUDA添加环境变量
打开设置,然后搜索编辑系统环境变量
打开,然后选择环境变量,选择path,选择编辑,新建变量,浏览找到
C:Program FilesNVIDIA GPU Computing ToolkitCUDAv11.1
C:Program FilesNVIDIA GPU Computing ToolkitCUDAv11.1libx64
这两个文件,添加进去
PS:v11.1是CUDA的版本,大家自行更改。
弄好之后,打开系统的命令提示符窗口
输入nvcc -V查看是否安装成功
这样就安装成功了,这里显示我的版本是v11.1
PS: 注意区分大小写,我一开始输入没有区分大小写输入了nvcc -v 一直显示出错,给我整懵了
(原谅我的英语有点渣,看不懂给的提示 qwq)
2 Anaconda在Anaconda官网上选择相应的版本进行下载
这边是我下载的版本
打开之后就是这样的,点击next
PS:这里要选一个空的文件夹哦
PS:这边建议全部勾选,不然后面配置会很麻烦
然后点击install安装就行了
安装好之后就会有这三个东西啦
然后打开Anaconda prompt 搭建一个属于自己的环境
打开之后输入
conda create -n pink_moon python=3.8
就可以搭建环境了
PS1:这里的pink_moon 是我给环境取的名,大家可以根据喜好更改
PS2:这里的3.8是我安装的python的版本,大家要根据自己的python的版本进行更改
摁下enter键,就会搭建啦
下面输入y,同意安装相关的拓展包
安装完毕之后
然后进入刚刚建立的环境
输入activate pink_moon(你自己的环境名) 进入环境
然后就ok啦
3 pytorch登录pytorch官网,选择相应的版本点进去。
我选择的是这个 CUDA 11.1的版本
复制pip代码,输入到Anaconda prompt中进行安装
文件有点大,安装需要一点时间,建议安装的时候不要动,不然容易出问题
安装完成后,重新打开Anaconda prompt
然后输入python
然后后输入
mport torch
print(torch.cuda.is_available())
就可以验证是否安装了cuda
返回True,cuda就安装好了
再输入
from torch.backends import cudnn
print(cudnn.is_available())
就可以验证是否安装了cuDNN
返回True,cuda就安装好了
像我这样就ok啦
PS:我这边是安装在base环境下的(大家可以进自己的环境去看看)



