2023最新pytorch快速安裝指南(超詳細(xì)版)
前言
因?yàn)槲抑鞍惭b環(huán)境花費(fèi)了不少時(shí)間，也是網(wǎng)上搜了很多，想著自己寫(xiě)一個(gè)幫助大家快速安裝，少走彎路，快速開(kāi)啟深度之路。
一、判斷是否有Nvidia(英偉達(dá)顯卡)
下面我以我的臺(tái)式電腦windows10系統(tǒng)為例
第一步：右鍵開(kāi)始菜單，在彈出選項(xiàng)中選擇任務(wù)管理器
進(jìn)來(lái)就是這個(gè)頁(yè)面，這臺(tái)是沒(méi)有GPU的
這臺(tái)是有GPU的
首先點(diǎn)性能選項(xiàng)，然后點(diǎn)GPU。在右上方會(huì)顯示GPU名稱(chēng)，只有帶NVIDIA的英偉達(dá)顯卡的電腦才能安裝GPU版本，否則其他的就只能安裝CPU版本。我會(huì)寫(xiě)兩部分，各位根據(jù)自己的電腦選擇觀(guān)看就好
二、CPU版
2.1 安裝Anaconda
首先去官網(wǎng)下載Anaconda。
點(diǎn)擊download，下載的就是最新版本的。
下載完成后，直接運(yùn)行下步就行
注意到路徑這塊，最好是全英文。
接下來(lái)根據(jù)個(gè)人喜好勾選，可以不用管直接下一步安裝好就行。
那怎么驗(yàn)證是否安裝好了呢？
在開(kāi)始菜單下，找到帶有綠色圖標(biāo)的Anaconda,點(diǎn)擊就可以打開(kāi)Anaconda的圖形化界面
點(diǎn)擊環(huán)境，看到base（root），這個(gè)就是默認(rèn)的一個(gè)虛擬環(huán)境，代表我們安裝成功了。
2.2 創(chuàng)建虛擬環(huán)境
為什么需要學(xué)會(huì)這個(gè)呢？其實(shí)剛剛安裝好Anaconda會(huì)自帶一個(gè)基礎(chǔ)環(huán)境。但是我們后續(xù)的項(xiàng)目每一個(gè)需要的安裝包不同，為了避免沖突，所以我們可以為每一個(gè)項(xiàng)目配置一個(gè)虛擬環(huán)境，這樣就不相互打擾了。
上面兩個(gè)是Anaconda命令窗口的位置，隨便點(diǎn)擊那個(gè)都可以。
conda env list
輸入上面的命令就可以查看當(dāng)前有多少虛擬環(huán)境。
• 利用 conda create 指令創(chuàng)建新的虛擬環(huán)境
? conda create –n 虛擬環(huán)境名字 python=版本
演示如下：
會(huì)出現(xiàn)如下圖
輸入y就行，然后就是等待安裝。好了以后用指令檢查一下，如下，就是成功的。
2.3安裝pytorch
等等，安裝前這里還有一個(gè)問(wèn)題，因?yàn)槲覀兊陌惭b指令都是從國(guó)外的通道下載，下載速度會(huì)非常慢。
所以我們想要下載快還是得用國(guó)內(nèi)的鏡像地址。
conda config --show
輸入上面的指令可以查看我們的通道地址
現(xiàn)在我們的環(huán)境中通道是defaults，也就是說(shuō)它會(huì)從國(guó)外給我下載包。
也可以通過(guò)下面的指令看到
conda config --get
下面是一些鏡像地址
| 鏡像名 | 用于創(chuàng)建環(huán)境鏡像地址 |
|---|---|
| 清華鏡像 | https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main |
| 北京外國(guó)語(yǔ)大學(xué)鏡像 | https://mirrors.bfsu.edu.cn/anaconda/pkgs/main |
| 阿里巴巴鏡像 | http://mirrors.aliyun.com/anaconda/pkgs/main |
關(guān)于加通道地址這塊，推薦是下載時(shí)可以指定安裝通道，而不是直接去改配置文件，這樣出現(xiàn)bug的幾率比較小。下面我也會(huì)提供這種方法不過(guò)不推薦。
- 往配置文件中加通道地址
conda config --add channels 通道地址
那么如何驗(yàn)證是否加上呢？
還是用上面的get指令
這里也會(huì)標(biāo)注，優(yōu)先使用我們新加上的通道，如果新通道沒(méi)有，再使用默認(rèn)通道去找。
- 萬(wàn)一你想刪除剛剛的通道怎么辦呢？用下面這個(gè)指令
conda config --remove channels 通道地址
- 安裝pytorch前想清楚一個(gè)問(wèn)題，想把pytorch安裝到那個(gè)環(huán)境里。
默認(rèn)是在基礎(chǔ)環(huán)境下（base），比如我現(xiàn)在想安裝到y(tǒng)olov8這個(gè)虛擬環(huán)鏡
conda activate yolov8
- 輸入上面命令進(jìn)入想安裝pytorch的虛擬環(huán)境，上面就是想在yolov8這個(gè)虛擬環(huán)境中安裝pytorch
進(jìn)入pytorch官網(wǎng)
conda install pytorch torchvision torchaudio cpuonly -c pytorch//這種就是直接從國(guó)外下載了
下面這種是鏡像加速版
conda install pytorch torchvision torchaudio cpuonly –c 鏡像地址
以清華鏡像源為例：
出現(xiàn)下圖，輸入y即可
出現(xiàn)下圖就是安裝好了
2.4 驗(yàn)證pytorch是否安裝成功
激活對(duì)應(yīng)的虛擬環(huán)境（你安裝Pytorch的虛擬環(huán)境）
conda activate 虛擬環(huán)境名
輸入conda list，看有沒(méi)有pytorch或者torch
比如我這里就是pytorch
輸入python
輸入 import torch
輸入 torch.cuda.is_available()，這個(gè)命令是檢查我們電腦是否可以用GPU，因?yàn)闆](méi)有，當(dāng)然就是false。
如果顯示False，就說(shuō)明我們這個(gè)PyTorch安裝成功了
三、GPU版
3.1 安裝Anaconda
首先去官網(wǎng)下載Anaconda。
點(diǎn)擊download，下載的就是最新版本的。
下載完成后，直接運(yùn)行下步就行
注意到路徑這塊，最好是全英文。
接下來(lái)根據(jù)個(gè)人喜好勾選，可以不用管直接下一步安裝好就行。
那怎么驗(yàn)證是否安裝好了呢？
在開(kāi)始菜單下，找到帶有綠色圖標(biāo)的Anaconda,點(diǎn)擊就可以打開(kāi)Anaconda的圖形化界面
點(diǎn)擊環(huán)境，看到base（root），這個(gè)就是默認(rèn)的一個(gè)虛擬環(huán)境，代表我們安裝成功了。
3.2 創(chuàng)建虛擬環(huán)境
為什么需要學(xué)會(huì)這個(gè)呢？其實(shí)剛剛安裝好Anaconda會(huì)自帶一個(gè)基礎(chǔ)環(huán)境。但是我們后續(xù)的項(xiàng)目每一個(gè)需要的安裝包不同，為了避免沖突，所以我們可以為每一個(gè)項(xiàng)目配置一個(gè)虛擬環(huán)境，這樣就不相互打擾了。
上面兩個(gè)是Anaconda命令窗口的位置，隨便點(diǎn)擊那個(gè)都可以。
conda env list
輸入上面的命令就可以查看當(dāng)前有多少虛擬環(huán)境。
• 利用 conda create 指令創(chuàng)建新的虛擬環(huán)境
? conda create –n 虛擬環(huán)境名字 python=版本
演示如下：
會(huì)出現(xiàn)如下圖
輸入y就行，然后就是等待安裝。好了以后用指令檢查一下，如下，就是成功的。
3.3 CUDA的準(zhǔn)備工作
要想使用GPU得先了解自己電腦的GPU。
鼠標(biāo)右鍵開(kāi)始菜單，點(diǎn)擊任務(wù)管理器，進(jìn)入性能中查看自己電腦GPU型號(hào)。
首先安裝最新的顯卡驅(qū)動(dòng),進(jìn)入官網(wǎng)選擇適合自己電腦的配置，我的配置如下。
下載完直接安裝就行。
然后是查看自己電腦驅(qū)動(dòng)的版本，win+r，輸入cmd，再輸入下面的命令
nvidia-smi
我的CUDA Driver版本就是11.6，表示的是驅(qū)動(dòng)所能支持的最大運(yùn)行API版本就是11.6。我如果要安裝CUDA Runtime Version（運(yùn)行版本），要保證CUDA Driver 版本 >= CUDA Runtime 版本，也就是11.6及以前的。
其實(shí)裝了Anaconda之后Anaconda會(huì)提供一個(gè)cudatoolkit工具包，同樣包含了CUDA的運(yùn)行API，可以用來(lái)替代官方CUDA的CUDA Toolkit。查看官方CUDA的運(yùn)行API版本。
nvcc -V
可以看到是11.1，也就是說(shuō)我的電腦安裝pytorch時(shí)最好選擇11.1-11.6版本的cuda。
3.4 安裝pytorch
等等，安裝前這里還有一個(gè)問(wèn)題，因?yàn)槲覀兊陌惭b指令都是從國(guó)外的通道下載，下載速度會(huì)非常慢。
所以我們想要下載快還是得用國(guó)內(nèi)的鏡像地址。
conda config --show
輸入上面的指令可以查看我們的通道地址
現(xiàn)在我們的環(huán)境中通道是defaults，也就是說(shuō)它會(huì)從國(guó)外給我下載包。
也可以通過(guò)下面的指令看到
conda config --get
下面是一些鏡像地址
| 鏡像名 | 用于創(chuàng)建環(huán)境鏡像地址 |
|---|---|
| 清華鏡像 | https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main |
| 北京外國(guó)語(yǔ)大學(xué)鏡像 | https://mirrors.bfsu.edu.cn/anaconda/pkgs/main |
| 阿里巴巴鏡像 | http://mirrors.aliyun.com/anaconda/pkgs/main |
添加通道
conda config --add channels 通道地址
那么如何驗(yàn)證是否加上呢？
還是用上面的get指令
這里也會(huì)標(biāo)注，優(yōu)先使用我們新加上的通道，如果新通道沒(méi)有，再使用默認(rèn)通道去找。
萬(wàn)一你想刪除剛剛的通道怎么辦呢？如下
conda config --remove channels 通道地址
關(guān)于加通道地址這塊，推薦是下載時(shí)可以指定安裝通道，而不是直接去改配置文件，這樣出現(xiàn)bug的幾率比較小。不推薦直接去改配置文件這種方法。
安裝pytorch前想清楚一個(gè)問(wèn)題，想把pytorch安裝到那個(gè)環(huán)境里。
默認(rèn)是在基礎(chǔ)環(huán)境下（base），比如我現(xiàn)在想安裝到y(tǒng)olov8這個(gè)虛擬環(huán)鏡
conda activate yolov8
輸入上面命令進(jìn)入想安裝pytorch的虛擬環(huán)境，上面就是想在yolov8這個(gè)虛擬環(huán)境中安裝pytorch
進(jìn)入pytorch官網(wǎng)
3.4.1 conda安裝法（不推薦）
進(jìn)來(lái)發(fā)現(xiàn)cuda不符，就去以前的版本找。
大家根據(jù)自己的環(huán)境需求找，我最后選擇這個(gè)。如果網(wǎng)絡(luò)還可以，就用官網(wǎng)的下載就行。
conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=11.3 -c pytorch
下面介紹一種鏡像的下載。
第一步：因?yàn)閏udatoolkit下載很慢，所以也單獨(dú)指定一個(gè)鏡像
Conda install cudatoolkit=版本 –c 鏡像地址
| 鏡像名 | 用于創(chuàng)建環(huán)境鏡像地址 - cudatoolkit |
|---|---|
| 清華鏡像 | https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main |
| 北京外國(guó)語(yǔ)大學(xué)鏡像 | https://mirrors.bfsu.edu.cn/anaconda/pkgs/main |
| 阿里巴巴鏡像 | http://mirrors.aliyun.com/anaconda/pkgs/main |
第二步：
conda install pytorch torchvision torchaudio -c 鏡像地址
| 鏡像名 | 鏡像地址 – pytorch, torchvision, torchaudio |
|---|---|
| 清華鏡像 | https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/ |
| 北京外國(guó)語(yǔ)大學(xué)鏡像 | https://mirrors.bfsu.edu.cn/anaconda/cloud/pytorch/win-64/ |
| 阿里巴巴鏡像 | http://mirrors.aliyun.com/anaconda/cloud/pytorch/win-64/ |
| 南京大學(xué)鏡像 | https://mirror.nju.edu.cn/pub/anaconda/cloud/pytorch/win-64/ |
這樣后續(xù)就好了。
3.4.2 pip安裝法（推薦）
進(jìn)入pytorch官網(wǎng)
如果你的電腦可以裝11.7或者11.8就直接粘貼復(fù)制到anaconda窗口，不行就像我一樣去以前的版本找。
3.5 驗(yàn)證pytorch是否安裝成功
激活對(duì)應(yīng)的虛擬環(huán)境（你安裝Pytorch的虛擬環(huán)境）
conda activate 虛擬環(huán)境名
輸入pip list或者conda list，看有沒(méi)有pytorch或者torch
比如我這里就是torch
3. 輸入python
4. 輸入 import torch
6. 輸入 torch.cuda.is_available()，這個(gè)命令是檢查我們pytorch的GPU能否用。
7. 如果顯示True，就說(shuō)明我們這個(gè)PyTorch安裝成功了
四、 pycharm安裝與配置
首先進(jìn)入pycharm官網(wǎng)
點(diǎn)擊download
下載社區(qū)版就夠用了。
注意路徑全英文。
我只勾了.py,其他的根據(jù)自己的需要來(lái)勾選。直接下一步等待安裝完成。
裝完后在開(kāi)始菜單下，找到pycharm點(diǎn)開(kāi)。
我喜歡亮一點(diǎn)，在這里調(diào)背景色。
開(kāi)始新建項(xiàng)目
自己指定項(xiàng)目的位置。
因?yàn)槲覀兦懊嬉呀?jīng)配好了環(huán)境，直接用我們配好的，點(diǎn)擊add去找我們配置的位置
從左往右按順序點(diǎn)擊，找到自己anaconda安裝的位置。
用我們已經(jīng)創(chuàng)建好的環(huán)境，比如我這里是yolov8
點(diǎn)擊create，項(xiàng)目就創(chuàng)建好了。
創(chuàng)建一個(gè)文件測(cè)試一下
import torch print(torch.cuda.is_available())
鼠標(biāo)右健后，點(diǎn)擊run就行
這就代表成功了，注意CPU版本這里應(yīng)該是False。
總結(jié)
到此這篇關(guān)于2023最新pytorch快速安裝指南的文章就介紹到這了,更多相關(guān)最新pytorch安裝內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！
相關(guān)文章
python經(jīng)典百題之畫(huà)圓形多種解決辦法
在Python中,您可以使用各種庫(kù)和工具來(lái)繪制圖形,其中包括繪制圓形,下面這篇文章主要給大家介紹了關(guān)于python經(jīng)典百題之畫(huà)圓形的多種解決辦法,文中通過(guò)代碼介紹的非常詳細(xì),需要的朋友可以參考下2024-03-03
Python中判斷subprocess調(diào)起的shell命令是否結(jié)束
這篇文章主要介紹了Python中判斷subprocess調(diào)起的shell命令是否結(jié)束的方式，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2021-04-04
python中將函數(shù)賦值給變量時(shí)需要注意的一些問(wèn)題
變量賦值是我們?cè)谌粘ｉ_(kāi)發(fā)中經(jīng)常會(huì)遇到的一個(gè)問(wèn)題，下面這篇文章主要給大家介紹了關(guān)于python中將函數(shù)賦值給變量時(shí)需要注意的一些問(wèn)題，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面來(lái)一起看看吧。2017-08-08
這篇文章主要為大家詳細(xì)介紹了如何利用Python語(yǔ)言實(shí)現(xiàn)超快窗口截圖功能，可以自動(dòng)獲取當(dāng)前活動(dòng)窗口并展示截圖，感興趣的可以了解一下2022-05-05
munch是Bunch包的分支，提供類(lèi)似的功能。99％的工作都是由他完成的，而前叉的創(chuàng)建主要是因?yàn)閷?duì)原始代碼的修復(fù)和維護(hù)缺乏響應(yīng)能力。Munch是支持屬性樣式訪(fǎng)問(wèn)的字典，本文將講解munch庫(kù)的使用2021-05-05
這篇文章主要為大家介紹了python內(nèi)置函數(shù)zip，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下，希望能夠給你帶來(lái)幫助2022-01-01
使用Python判斷質(zhì)數(shù)(素?cái)?shù))的簡(jiǎn)單方法講解
這篇文章主要介紹了使用Python判斷質(zhì)數(shù)(素?cái)?shù))的簡(jiǎn)單方法講解,經(jīng)常被用來(lái)做科學(xué)計(jì)算的Python處理這種小問(wèn)題當(dāng)然手到擒來(lái)^_-需要的朋友可以參考下2016-05-05
super 是用來(lái)解決多重繼承問(wèn)題的，直接用類(lèi)名調(diào)用父類(lèi)方法在使用單繼承的時(shí)候沒(méi)問(wèn)題，但是如果使用多繼承，會(huì)涉及到查找順序（MRO）、重復(fù)調(diào)用（鉆石繼承）等種種問(wèn)題。這篇文章主要給大家介紹了關(guān)于Python中super()方法的相關(guān)資料，需要的朋友可以參考下。2017-11-11
Python對(duì)數(shù)據(jù)進(jìn)行插值和下采樣的方法
今天小編就為大家分享一篇Python對(duì)數(shù)據(jù)進(jìn)行插值和下采樣的方法，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2018-07-07
基于Python實(shí)現(xiàn)的微信好友數(shù)據(jù)分析
這篇文章主要介紹了基于Python實(shí)現(xiàn)的微信好友數(shù)據(jù)分析的相關(guān)知識(shí)，非常不錯(cuò)，具有參考借鑒價(jià)值,需要的朋友可以參考下2018-02-02