Conda 离线安装pytorch-nightly

2020-06-01  本文已影响0人  昵称己存在

国内conda安装pytorch的时候回很慢。通过清华源可以再一定程度上解决这个问题,但是由于pytorch-nightly版本跟新太快,清华源没有pytorch-nightly版本。 本文介绍如何离线安装pytorch-nightly

创建环境的步骤省略。

区别在于online安装,只需要

conda install pytorch torchvision cudatoolkit=10.2 -c pytorch-nightly

离线安装,需要先把对应的包下载到本地。
网址:https://anaconda.org/pytorch-nightly/repo

下载自己需要的pytorch, torchvision版本。


image.png

如图,选中的是cuda10.2-python3.7版本。直接浏览器下载得到*.tar.bz2文件

最后通过

conda install --offline xxxx.tar.gz2

安装好pytorch, torchvision。

cudatoolkit=10.2可以直接通过:

conda install cudatoolkit=10.2

安装。

Best!

上一篇下一篇

猜你喜欢

热点阅读