WIN10环境下安装pytorch:
尝试安装gpu版本的pytorch,尝试无数次以后,终于成功。
安装步骤:
1.首先查找自己gpu版本信息,在cmd中输入:
nvidia-smi
得到版本信息
我这里的CUDA支持11.1,在去pytorch官网看到可以安装10.1版本。
2.conda prompt中创建pytorch环境。
conda create --name torchenv python=3.6
3.激活创建的环境
activate torchenv
4.改掉conda安装的镜像源,将以下复制进.condarc文件中。
channels:
- http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
- http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/