一、安装Anaconda
下载pytorch前最好先安装Anaconda,可以按照以下方法下载:
https://blog.csdn.net/weixin_50888378/article/details/109022585
二、安装pytorch
1、安装好Anaconda后,在电脑搜索栏搜索Anaconda,点击进入Anaconda Prompt。
2、换源
在Anaconda终端输入:
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/
像这样:
3、创建python3.7的虚拟环境,名叫pytorch
在anaconda的终端里输入:
conda create -n pytorch python=3.7
出现下图这个选项时,这里选y
4、使清华源可见
输入 conda config --set show_channel_urls yes
5、执行红框中的 conda activate pytorch
如图
6、由于不翻墙下载太慢,这里选择离线下载,下载包可到官网查看对应版本:
官网查看pytorch与torchvision对应版本:https://pytorch.org/get-started/previous-versions/
下载地址:https://download.pytorch.org/whl/torch_stable.html
下载完成后,将三个包放入终端显示的当前文件夹下(如我的终端显示的是c盘->Users(用户)->86151)
7、在Anaconda终端依次输入:
pip install 你下载的torch名
pip install torchaudio-0.7.2-cp37-none-win_amd64.whl(你下载的torchaudio名)
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
pip install torchvision-0.8.2+cpu-cp37-cp37m-win_amd64.whl(你下载的torchvision名)
注意:最好用移动数据下载,wlan可能会很慢。
到此为止pytorch安装完毕。在这里感谢QQ里某位大佬,这是他的安装方法,我只是将这个方法记录下来。
在pycharm里可能会出现找不到torchvision的错误问题,是因为没有设置pycharm的环境 可通过https://www.cnblogs.com/jaysonteng/p/12554161.html解决。