PyTorch框架,據說2018.4.25剛剛上架windows,安個玩玩
我的環境:
windows 10
anaconda虛擬環境python3.6
cuda9.1
cudnn7
pytorch 0.4.0
cuda9和cudnn我之前安裝tensorflow的時候已經安裝過了,這里接着在tensorflow的環境下安裝,省的再安裝其他模塊了
1.whl下載
官網上和別人的教程都是cmd或conda下載安裝,我試了試慢的不行,還出錯,所以就單獨下載whl文件,下載地址http://download.pytorch.org/whl/cu91/torch-0.4.0-cp36-cp36m-win_amd64.whl
這個都是對應我的版本的,其他版本的
2.常規安裝
(1)先查看一下虛擬環境,tensorflow-gpu是我3.6的環境,激活一下
(2)切換到whl文件位置安裝
(3)successful,測試一下子
(4)最后安裝torchversion,我這用的校園網,看外網有點費勁,就讓別人幫我下的whl
還是3.6環境下切換到whl文件所在地址,安裝
完 美~