anaconda环境下安装pytorch

环境：win7
安装思路：创建虚拟环境 -> 切换到虚拟环境下 -> 安装pytorch

1.创建虚拟环境

// your_env_name 你的虚拟环境名
// python=X.X 指定版本号
conda create -n your_env_name python=X.X

2.切换到虚拟环境下

activate your_env_name

3.添加清华源镜像

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes

4.到官网根据python和CUDA选择对应的版本

https://pytorch.org/get-started/locally/

image.png

由于我的CUDA版本过低，所以在CUDA中选择None，注意在安装时一定要去除 -c pytorch，这样才能通过清华源镜像加速安装

conda install pytorch torchvision cpuonly 

5.检查是否安装完成

image.png

6.踩过的坑

1.版本问题，在安装时一定要先确认好版本问题，大多数问题都源于版本问题
2.创建虚拟环境时，代码出错，输入 conda create -n python=X.X 没有环境名，只有路径，删除路径重新创建虚拟环境

conda env remove -p  路径名