打算把項目的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相應目錄文件夾里面就好。 大功告成