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
中的命令下载安装
使用图片中的命令结果报错:
再结合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中的验证:
上一篇: Qt在Mac环境下制作dmg安装包