天天看點

一分鐘搞懂在Windows10系統安裝使用pytorch架構

該文章将記錄如何在Windows10系統安裝制定版本的pytorch架構,以便使用上電腦的獨立顯示卡GPU。

第一步,安裝cudn和cudnn,内容較長請看我的另外一篇部落格文章。

注意,從“第三件事步驟”開始安裝就行。我自己安裝的版本是cudn9.0和cudnn7.1版本。該篇文章也會以這個版本為例。

第二步,下載下傳pytorch。如果線上安裝的話慢得懷疑人生,是以推薦先下載下傳好安裝包再進行安裝。安裝包下載下傳連結如下:https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/

打開之後挑選适合自己的安裝包

一分鐘搞懂在Windows10系統安裝使用pytorch架構

第三步,安裝pytorch,安裝語句:

conda install --offline -n deep_image_matting2 pytorch-1.1.0-py3.6_cuda90_cudnn7_1.tar.bz2語句解釋:–offline表示離線安裝。-n後面接的是虛拟環境的名字。最後面是安裝包的目錄檔案名

一分鐘搞懂在Windows10系統安裝使用pytorch架構

第四步,安裝pytorch的依賴包torchvision。pytorch為1.1.0版本的時候,torchvision對應的版本為0.3.0.具體的版本比對資訊可以看下面連結的說明:https://pytorch.org/get-started/previous-versions/

安裝語句為:pip install -i https://pypi.tuna.tsinghua.edu.cn/simple torchvisinotallow==0.3.0

第五步,其實完成到這一步就完成所有事情了,可是個人電腦出現Pillow版本不對的現象(找不到Pillow Version),是以我就解除安裝了Pillow7.0.0版本的Pillow而重新安裝了Pillow5.3.0版本的Pillow.附上解除安裝和安裝的語句:

pip uninstall Pillow

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple Pillow==5.3.0

第六步,驗證一下是否正确。看截圖

一分鐘搞懂在Windows10系統安裝使用pytorch架構

繼續閱讀