Install pytorch on windows7

在win下運行,會找不到pytorch和torchvision(如圖所示):python

安裝pytorch的方法,是在這個知乎專欄裏找到的,附上地址,以示對做者的尊重:windows

https://zhuanlan.zhihu.com/p/26871672spa

安裝的命令有四個,第一個表示沒有cuda使用pytorch,也就是隻有cpuget

其餘是安裝了對應版本的cuda以後,安裝pytorch 的命令(忽略掉註釋裏的windows版本,只須要看cuda對應的版本就好~,個人cuda版本是9.1.85,最新版本,選擇的是第三個命令,安裝可用,都是屬於cuda9),在安裝的時候,最好先牆出去,速度會快些,固然不牆也不要緊,就是等的久一點而已嘛~io

這樣,pytorch就安裝完畢了,可是還有pytorchvision須要安裝,在使用時,pytorch和pytorchvision每每須要同時出現,pytorchvision安裝方法,我是經過本地安裝的方式進行安裝的,本地包的下載地址爲:https://pypi.python.org/pypi/torchvision/0.1.8
下載

下載好以後,在當前文件夾下打開命令窗,輸入下面的命令,安裝pytorchvision:方法

至此,全部的安裝就已經完成。im

下面查看是否安裝好pytorch,torchvision和pytorch是否支持cuda,輸出爲True,證實ok。專欄

相關文章
相關標籤/搜索