電腦里以前安裝了 tensorflow,現在因為學習需要,需要安裝pytorch。還是在原來安裝tensorflow的位置安裝pytorch。
由於采用在線安裝太慢了,而且中途還會因為網速不穩定終端!因此,采用離線安裝(真的是快啊),先將具體操作記錄如下
(PS:本文只講在已經安裝CUDA和CUDNN等一系列前提下,進行的)
1、首先去清華鏡像下載pytorch
https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/
找到對應的版本下載即可,本人安裝的是
2、打開anaconda prompt,執行如下操作
(base) D:\software\Anaconda3>conda install --offline -n base pytorch-nightly-1.0.0.dev20190101-py3.6_cuda90_cudnn7_0.tar.bz2
(注意 base是默認的虛擬環境,回車等待安裝即可)
3、安裝pytorch的支持庫 torchvision
pip3 install --no-deps torchvision
4、測試
注意: 在安裝torchvision的時候遇到的錯誤
解決辦法,將pip install torchvision 改為 pip3 install --no-deps torchvision 即可解決!