首先更换清华源
- 终端中输入:gedit ~/.condarc
- 将内容更改为:
channels:
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/peterjc123
- defaults
show_channel_urls: true
安装
- 如果已经安装虚拟环境,记得先打开虚拟环境。虚拟环境的安装可见:https://blog.youkuaiyun.com/JianJuly/article/details/80300473
- 安装链接
- 在安装链接中,根据操作系统和
CUDA版本选择安装指令,并使用指令进行安装 - 注意事项:记得一定要把指令最后的 -c pytorch 删除,不然会强制使用默认的源,安装速度特别慢

最后
打开 python,输入以下内容测试并查看版本
import torch
import torchvision
print(torch.__version__)
print(torchvision.__version__)


本文介绍如何通过更换清华源来加速PyTorch的安装过程,包括修改配置文件~/.condarc的具体步骤,以及在不同操作系统和CUDA版本下选择合适安装指令的方法。文章还提醒在安装过程中删除-c pytorch以避免使用默认源,最后提供了测试安装是否成功的Python代码。
3076

被折叠的 条评论
为什么被折叠?



