参考《手把手教你安装pytorch(包教会)》
https://blog.csdn.net/ziqingnian/article/details/109099682
使用Conda安装:
打开 anaconda prompt,最前面的小括号()显示应该是base
(注意:在base的情况下安装pytorch则安装在默认环境下;如果新建了别的环境则需要先激活别的环境,然后在该环境下安装)
例如:conda create -n pytorch python=3.6 这里是创建了一个名称为 pytorch 环境,它的编译器python版本为3.6
可以使用命令conda activate pytorch激活该环境,然后在该环境下安装pytorch
如果不激活别的环境的话,默认在base下安装
添加清华镜像:
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
还需要添加pytorch的镜像
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
百度下载pytorch:
注意我们需要copy ‘-’ 符号之前的因为如果全部copy,会使用默认的下载而不是清华镜像下载所以假设上图,我应该复制 conda install pytorch torchvision cudatoolkit=10.1
然后等待安装完成就可以了
验证是否安装成功:
ENVS文件夹下面是创建的环境;Lib文件夹内是已经安装的各种包。
更改Python包安装源:
https://blog.csdn.net/qq_29007291/article/details/81103603