先确保已安装cuda(请点击查看是否安装了cuda),cudnn(请点击查看cudnn的安装),然后才能装pytorch。
一、安装pytorch
1、激活虚拟环境。
2、打开pytorch官网
https://pytorch.org/,如果版本合适,按下图中最底下的命令装。如果和自己的版本不一样,要把对应的版本改成自己的版本。并把-c pytorch后面的删除,这样就能在清华源中下载。
通过在终端输入nvcc -V,查询自己的cuda版本。
我的cuda是11.6,在官网首页上并没有相应的命令,通过点击首页下面的Previous versions of PyTorch。
可以看到之前的版本,及其相应的命令,它有pip安装和conda安装两种,可以根据自己电脑的配置,选择相应的版本,我选择的是1.12.0版本。
(1)conda安装时的命令