欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

Pytorch+torchvision的搭建

程序员文章站 2022-05-22 13:07:50
...

Step1:安装anaconda,具体见:

 

Step2:打开Anaconda Prompt,查看并且更新conda

#查看conda,发现不是最新版本
conda -V
#更新conda直到最新版本conda 4.8.3,需要更新2回,才能更新到最新版本
conda update conda 

Step3:接着更新pip

python -m pip install --upgrade pip

Step4:打开网址:https://pytorch.org/,选择符合自己电脑的配置,使用pip安装,因为conda太慢了,使用镜像也十分的慢。

这里面需要注意的是:我们使用图片里面的命令安装的时候,会十分的慢,所以我们要结合我们的报错信息,使用Step5、Step6

中的命令下载安装

Pytorch+torchvision的搭建

使用图片中的命令结果报错:

Pytorch+torchvision的搭建

再结合https://download.pytorch.org/whl/torch_stable.html,可以推断出来我们的包是Step5、Step6中的包

Step5:下载并且安装pytorch(即:torch)

pip install https://download.pytorch.org/whl/cu101/torch-1.5.0%2Bcu101-cp36-cp36m-win_amd64.whl -i http://mirrors.aliyun.com/pypi/simple/  --trusted-host mirrors.aliyun.com

Step6:下载并且安装torchvision

pip install https://download.pytorch.org/whl/cu101/torchvision-0.6.0%2Bcu101-cp36-cp36m-win_amd64.whl -i http://mirrors.aliyun.com/pypi/simple/  --trusted-host mirrors.aliyun.com

我们没有使用新得环境哈,完全是在base环境上安装的这两个包,所以我们在pycharm中验证的时候,文件的路径的设置要注意

最后在pycharm中的验证:

Pytorch+torchvision的搭建

相关标签: 杂文