1. 程式人生 > 其它 >差點放棄的win10 pytorch環境搭建

差點放棄的win10 pytorch環境搭建

技術標籤:pytorchpytorch

寫在最前面。千萬不要用線上下載搭建,千萬不要用線上下載搭建!千萬不要用線上下載搭建!!!

只要有人有一步說輸入“

conda install pytorch==1.6.0 torchvision==0.7.0 cudatoolkit=10.2 -c pytorch

”請離開,絕對是個騙子。不知道哪裡抄來的,估計pytorch是什麼東西他都不知道。

所謂的手把手教你搭建pytorch環境,最詳細搭建環境,只要是線上下載,都是扯淡。

用預設的環境搭建,下載很慢,修改清華源,還是很慢。開著翻牆,一點作用都沒有,我懷疑Facebook是故意不給國內提供下載。總之,conda指令很慢,浪費很多時間也搭不起來。

下面是我搭建成功的步驟。

第一,安裝cuda,我的是10.2,cudnn8.0.2,雖然pytorch給的對應版本很奇怪,比如安裝1.7.1,線上提醒安裝的是cuda10.2,cudnn7.0(cuda官網壓根就沒有對應10.2的cudnn7.0簡直是扯淡誤導),這個可以忽略,只要cuda跟cudnn對應,cuda跟pytorch對應即可

第二,安裝anaconda,版本可以忽略不計,因為可以自己建虛擬環境,安裝好anaconda之後搭建自己的虛擬環境,然後啟用,具體做法可以去看其他人的csdn,都可以用

第三,conda config --append channels conda-forge,

第四,conda install slycot control,

第三第四步是安裝了一些基礎包。

第五,最關鍵的跟其他人不一樣的,去這個網址下載對應的torch以及torchvision,https://download.pytorch.org/whl/torch_stable.html

第六,在conda命令框中,cd到這兩個資料夾所在目錄,利用pip install *****.whl,或者不用cd過去直接pip install 路徑/檔名即可

至此環境搭建完畢