Anaconda pytorch-gpu CUDA CUDNN 安装指南
安装教程漫天飞,博主此处记录安装的一些关键节点。
1、CUDA与CUDNN
首先对安装CUDA与CUDNN,建议可以按照此博客:http://t.csdnimg.cn/VkGm3
安装之前建议先看看自己要安装的pytorch-gpu所使用的CUDA版本。博主计算机是可以安装到CUDA12.6的,但是发现pytorch最高12.4,所以重新安装走了弯路。
2、Anaconda
官网下载,启动安装。基本没问题,有一些要注意的点:
1)选择为谁安装:建议选择为自己安装。为所有人安装后面为环境下载安装的时候会需要管理员权限。就需要以管理员启动conda的终端,有点麻烦其实。
2)安装位置:默认C盘个人文件下,但是可以考虑安装到其他盘。别犯浑劲试图将anaconda安装到类似C盘下的programfile或者programdata等文件夹中,不然也会出现各种事情都要管理员权限。
3)系统环境:如果是为所有人安装不会出现添加path的勾选项。如果为自己安装,就会出现path勾选框,然后下面的描述也说了:不建议,所以这个框尽量别勾选,后续安装完自己再去系统环境中添加。
3、pytorch-gpu
呐,这个就简单了。
先创建一个虚拟环境,终端创建还是Anaconda Navigator创建都行。打开对应环境终端。
然后去官网,图就和上面第一幅一样选择好后,把命令输入到终端,等待安装完毕即可。
有各种方式测试安装效果。给出一种这里,直接在安装完的终端输入python,然后
import torch
torch.get_default_dtype()