前期准备:

1、配置python环境,python3.8
2、安装Anaconda最新版(建议通过清华镜像下载安装包)
3、配置CUDA11.4(前提是电脑有NVIDIA显卡)

安装pytorch:

(1)网上好多推荐使用Conda,先添加清华镜像路径

conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge 
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

# 设置搜索时显示通道地址
conda config --set show_channel_urls yes

可以在C盘用户目录下找到.condarc的文件,内容如下所示,如果带有—defualts务必删掉,还要确保都是http而不是https(不太懂为什么,不过改过来就可以正常下载)

channels:
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/

show_channel_urls: true

remote_read_timeout_secs: 600.0

进入pytorch官网,选择合适的安装命令
运行命令
以最新版本为例,管理员身份进入cmd,输入如下命令(因为使用清华镜像下载,所以不要带-c pytorch)

conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c conda-forge

(2)使用pip下载安装pytorch
进入pytorch官网,选择合适的安装命令
在这里插入图片描述
以最新版本为例,管理员身份进入cmd,输入如下命令

pip3 install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio===0.9.0 -f https://download.pytorch.org/whl/torch_stable.html -i  https://pypi.tuna.tsinghua.edu.cn/simple

后面加的-i https://pypi.tuna.tsinghua.edu.cn/simple就表示通过清华镜像下载。

测试安装结果:

在cmd中输入

python

进入python编译器,输入

import torch
torch.__version__

输出为相应的版本号:

'1.9.0+cu111'
Logo

为开发者提供学习成长、分享交流、生态实践、资源工具等服务,帮助开发者快速成长。

更多推荐