國內(nèi)conda安裝pytorch的時(shí)候回很慢。通過清華源可以再一定程度上解決這個(gè)問題,但是由于pytorch-nightly版本跟新太快,清華源沒有pytorch-nightly版本。 本文介紹如何離線安裝pytorch-nightly
創(chuàng)建環(huán)境的步驟省略。
區(qū)別在于online安裝,只需要
conda install pytorch torchvision cudatoolkit=10.2 -c pytorch-nightly
離線安裝,需要先把對(duì)應(yīng)的包下載到本地。
網(wǎng)址: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!