方法一
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
出现上图所示即为安装成功。
测试是否安装成功
import torch
x = torch.rand(5, 3)
print(x)
结果如下所示
方法二
1、安装anocanda
网上有教程,这里不再说明
2、安装pytorch
进入pytorch的官网进行版本选择 网址pytorch.org/get-started…
这里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