1、参看查询CUDA版本号
打开计算机终端(win+R,输入cmd,即可进入),输入:
nvidia-smi
cuda版本如下:

注:记住自己cuda版本,后续需要!
2、查看如需的python版本
前往pytorch官网:PyTorch,选择相应的GPU版本。根据自己的 cudaversion的版本选择相应的pytorch版本。红色框处显示需要3.9或3.9以上的python版本

3、下载python版本
进入到python下载,根据2、下载并安装相应的python版本

安装python时,记得勾选红色框选项

4、下载并安装vscode
5、安装 miniconda
win+r输入cmd打开终端,输入command prompt下的指令。红色方框所示。
如果找不到miniconda软件,表示只是下载了其安装包并没有进行安装。在开始处输入miniconda,找到相应的安装包并进行安装。
在菜单中能够看到anaconda文件夹,则表示minianaconda安装完成。如下图所示:

6、安装cuda与cudnn
6.1 下载及安装cuda
在安装pytorch之前需安装cuda,下载cuda前需要先查看显卡支持的CUDA版本最高是多少,按下win+r键,输入cmd,在打开的页面输入:nvidia-smi ,即可查看。
在这个网站根据自己的显卡版本进行下载。下载完之后运行安装,建议安装到默认路径,所以C盘需要留有20G以上的存储空间,一直点击下一步,直到出现这个界面点击自定义,然后全部勾选即可。
安装步骤如下:
step1:打开官网,点击与自己显卡对应的cuda版本。



最低0.47元/天 解锁文章
912

被折叠的 条评论
为什么被折叠?



