Pytorch開(kāi)發(fā)環(huán)境搭建

以windows作為開(kāi)發(fā)環(huán)境
本人的筆記本配置情況:
CPU:AMD R7-6800H
顯卡:RTX 3060?
系統(tǒng):Windows 11
安裝好Windows和必要的驅(qū)動(dòng)后,需要安裝的工具有:CUDA、Anaconda3、cuDNN、Pytorch-gpu
1. CUDA
CUDA是NVIDIA推出的運(yùn)算平臺(tái)。我們需要根據(jù)自己的顯卡型號(hào)選擇支持的CUDA版本,例如RTX 3060支持CUDA 12,下載地址:https://developer.nvidia.com/cuda-downloads?target_os=Windows&target_arch=x86_64&target_version=11&target_type=exe_local

傻瓜式安裝。
安裝好后,需要再配置下系統(tǒng)的環(huán)境變量,電腦->鼠標(biāo)右鍵->屬性->高級(jí)系統(tǒng)設(shè)置->環(huán)境變量->Path:

這里我安的比較早,所以是CUDA11。

有時(shí)候傻瓜式安裝完了就行~~~~~~
配置好后,就可以在cmd中使用nvidia-smi指令查看顯卡了。
2. Anaconda3
Anaconda是Python的包管理器和環(huán)境管理器,可以方便我們安裝Python的第三方庫(kù)。

選擇Windows下的Python 3.9的版本,安裝也很簡(jiǎn)單,傻瓜式下一步即可。
安裝好后,需要添加系統(tǒng)環(huán)境變量,方法與安裝CUDA時(shí)一樣。
路徑改為自己安裝的Anaconda路徑即可。
配置好后,在cmd中運(yùn)行conda -V沒(méi)有報(bào)錯(cuò),有版本信息輸出,說(shuō)明配置成功。
3. cuDNN和Pytorch安裝
安裝好Anaconda之后,使用conda安裝cuDNN和Pytorch。
打開(kāi)Anaconda Prompt,這是Anaconda自帶的命令行工具,先一定要用這個(gè)工具創(chuàng)建環(huán)境,直接用系統(tǒng)自帶的cmd,可能遇到一些奇怪的問(wèn)題,例如CondaHTTPError的錯(cuò)誤。在Anaconda Prompt中輸入:
conda create -n your_name jupyter notebook
這句話(huà)的意思是創(chuàng)建一個(gè)名字為your_name的虛擬環(huán)境,并且這個(gè)虛擬環(huán)境額外安裝jupyter notebook第三方庫(kù)??梢詫our_name改為你自己喜歡的名字,這個(gè)名字是你的虛擬環(huán)境的名字,自己隨便取,比如LSP。
隨后,輸入y進(jìn)行安裝:

安裝好后,可以通過(guò)指令conda info -e查看已有環(huán)境情況:

從上圖可以看到,有四個(gè)環(huán)境,一個(gè)是base,自帶的基礎(chǔ)環(huán)境,其他三個(gè)是創(chuàng)建的新環(huán)境。新建環(huán)境可以分開(kāi)管理我們配置的環(huán)境。
安裝好環(huán)境后,我們就可以激活xzhy環(huán)境,并安裝cuDNN和GPU版的Pytorch了。激活名為xzhy的環(huán)境:activate xzhy

可以看到,我們的環(huán)境由base變成了xzhy。在xzhy環(huán)境中安裝cuDNN:conda install cudnn
安裝cuDNN好后,安裝Pytorch,打開(kāi)Pytorch官網(wǎng):https://pytorch.org/

根據(jù)自己的環(huán)境選擇,選擇好后,網(wǎng)頁(yè)會(huì)自動(dòng)給出需要運(yùn)行的指令。這里可能需要區(qū)分下Python的版本和CUDA的版本。
Python版本查看方法:直接在命令行中輸入python,會(huì)看到Python的版本。

CUDA版本查看方法,在命令行中輸入nvidia-smi:

確定好版本后,就可以通過(guò)Pytorch官網(wǎng)提供的指令安裝GPU版本的Pytorch了。
至此,基礎(chǔ)的環(huán)境搭建已經(jīng)完成。