之前发过一篇记录过程的仅记录,这次提供一个超快速版本,理论上是全平台适用。
-
确认你的机器是否支持cuda以及支持对应的最高版本
nvidia-smi
,如下图:
意味着只能安装11.6及11.6以下的版本 -
开始安装,首先创建虚拟环境,直接创建就可以了,如
conda create -n CSDN python==3.8
,这里-n
后面是你创建虚拟环境的名字,指定python版本为3.8.进入虚拟环境conda activate CSDN
可以看到新创建的虚拟环境下面几乎是什么都没有的,如下图:
-
部署pytorch及cuda环境,登陆pytorch官网,向下拉,找到INSTALL PYTORCH:
我的设备最高支持11.6版本的cuda,因此要进入下面的previous versions of PyTorch,安装符合你要求的版本,如下图:
只要是11.6及之前的版本都可以,可以用pip或conda安装不影响,个人推荐conda,复制完整的命令,粘贴到终端,安装完成后对应的cuda,cudnn就都安装好啦。这个方法非常快速,省略掉了安装cudnn时还需要官网注册等步骤,可以说是一键安装了,缺点是没创建一次虚拟环境都要重新装一次(不过输一行命令也不费什么事)