Win10+miniconda+cuda+cudnn+pytorch1.5.1安裝記錄(踩坑記錄)


打算把項目的tensorflow的項目裝換成pytorch的,打算在win10電腦上裝個環境。

1 安裝一個miniconda3,選擇miniconda的原因是因為這個方便。安裝完成之后,使用命令 

1 conda create -n hh

創建虛擬環境,其中上文的中的 hh 是虛擬環境的名稱,自己可以隨意,叫goudan都可以,python的版本隨意,反正后面都會被覆蓋。

2 創建環境完成之后,然后為創建的虛擬環境安裝pip 工具 conda install pip -n hh ,之所以為虛擬環境安裝pip,是因為使用conda安裝pytorch,根本裝不上,加上梯子也不行。

我這里是自動給安裝上了python3.8

3 安裝pytorch,首先進入虛擬環境 conda activate hh ,然后找到pytorch的官網,選擇合適的版本,我這里直接給鏈接了 https://pytorch.org/get-started/locally/ ,我這里使用的是cuda10.2版本的。復制下面的命令 ,然后運行。如果是cpu版本的,CUDA那里直接選擇None那一項,安裝就好,並且cpu版本的到這里就安裝結束,如果只是用來學習,cpu版本的pytorch就夠了(順便說一下,加了梯子下載速度超快)。

4 安裝cuda,最近從英偉達官網下載的cuda安裝包都是40多K的不知道什么東西,然后從別人網盤里面找到了安裝包,我也把鏈接貼到這里,我選擇的是10.2版本的,里面有10.1,10.2,11三個版本的

鏈接:https://pan.baidu.com/s/1yD4lyEPHcbzRA6x8krHRpg
提取碼:a81b

下載完成之后,一路next,我裝到C盤了,如果C盤空間不夠,裝到別的盤應該也沒有問題。

安裝完成之后,在命令行中輸入  nvcc -V ,有如下輸出 ,就表示沒有問題。

 

 

 5 安裝cudnn 這個包從官網下載就好,下載的版本要和已經安裝的cuda的版本對應,我下載的版本是7.6.5 for CUDA 10.2,

 

 

 

下載完成之后解壓縮,進入解壓好的文件,將文件夾里面的文件復制到CUDA相應目錄文件夾里面就好。 大功告成

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM