windows下anaconda中pyTorch安装

windows下anaconda中pyTorch安装

1.torch官网查询conda安装命令并运行

torch官网地址：

https://pytorch.org/

点击进去后，往下面拉，到下面的位置，选择自己电脑对应版本，此处以window、3060显卡为例（注意30系显卡cuda版本需大于11）

去掉命令后面的 -c pytorch，目的是借助第三方镜像源快速安装pytorch，如下

conda install pytorch torchvision torchaudio cudatoolkit=11.3 

新建环境，配置第三方镜像源，具体过程详见anaconda常用使用命令windows环境下anaconda常用(不定期更新) - 知乎 (zhihu.com)

在conda环境中运行上面的命令

安装过程

2.安装cudnn

看cuda和cudnn对应的版本

https://developer.nvidia.com/rdp/cudnn-archive

这里cudatoolkit的版本为11.3，选择cudnn的版本为8.2.0安装

或者在conda中搜索可以安装的cudnn版本，conda中输入命令如下

conda search cudnn

选择cudnn8.2.1安装，命令如下

conda install cudnn=8.2.1

安装过程

安装过程

[外链图片转存中...(img-UZ6sBIL4-1652778230262)]