镜像安装pytorch的简便方法

5,942 阅读1分钟

方法一

1、安装anocanda

网上有教程,这里不再说明

2、配置清华镜像源


conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/peterjc123/

conda config --set show_channel_urls yes

3、修改.condarc文件,删除里面的-defaults这一行

vi ~/.condarc 最后:wq!保存退出即可

4、执行安装

先创建虚拟环境

conda create -n torch python=3.6.4 根据自己的python版本输入,torch为虚拟环境名

激活环境

conda activate torch

执行安装命令

conda install pytorch torchvision

image.png 出现上图所示即为安装成功。

测试是否安装成功

import torch
x = torch.rand(5, 3)
print(x)

结果如下所示

image.png

方法二

1、安装anocanda

网上有教程,这里不再说明

2、安装pytorch

进入pytorch的官网进行版本选择 网址pytorch.org/get-started…

image.png

这里package选择pip

安装命令为

pip install torch===1.9.0 torchvision===0.10.0 -f https://download.pytorch.org/whl/torch_stable.html>

这里pytorch最新版本为1.9.0,所以这里以安装1.9.0为例

在这个安装命令后跟随镜像名称 -i https://pypi.tuna.tsinghua.edu.cn/simple some-package -i后面跟的是镜像名称了,some-package为固定格式。

以下是一些镜像源

豆瓣(douban) pypi.douban.com/simple/

清华大学 pypi.tuna.tsinghua.edu.cn/simple/

阿里云 mirrors.aliyun.com/pypi/simple…

中国科技大学 pypi.mirrors.ustc.edu.cn/simple/

中国科学技术大学 pypi.mirrors.ustc.edu.cn/simple/

清华大学和科大镜像源可能会出错,作者测试多次安装清华镜像源都失败了,最终采用阿里云镜像安装成功。

推荐采用阿里云的镜像源,速度最快。

最终的pip命令为

pip install torch===1.9.0 torchvision===0.10.0 -f https://download.pytorch.org/whl/torch_stable.html -i http://mirrors.aliyun.com/pypi/simple/ some-package

镜像安装会比官网,清华镜像速度快很多,以下就是pytorch正在安装的过程,只需静静等待安装成功,这个时间段最好不要让电脑息屏,不然会影响下载速度,可能会安装失败。

image.png