使用清华镜像安装pytorch
前期准备:1、配置python环境,python3.82、安装Anaconda最新版(建议通过清华镜像下载安装包)3、配置CUDA11.4(前提是电脑有NVIDIA显卡)安装pytorch:(1)网上好多推荐使用Conda,通过清华镜像安装下载pytorch先添加清华镜像路径conda config --add channels http://mirrors.tuna.tsinghua.edu.c
·
前期准备:
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'
更多推荐
已为社区贡献1条内容
所有评论(0)