pytorch gpu是必須裝的，因?yàn)閏pu太慢了。
最新的pytorch版本是2.0
下面是安裝教程。
- 首先，登錄pytorch.org網(wǎng)站，看你要安裝的cuda版本。
目前最新的支持cuda11.8，于是我選擇了11.8 - 然后，登錄cuda官網(wǎng)，下載cuda11.8
cuda最新版已經(jīng)是12.1了。pytorch不支持！！?。?！所以，找11.8版下載。 - 安裝cuda。
安裝的時候，gpu顯卡驅(qū)動需要用cuda11.8版本里面帶著的顯卡驅(qū)動?？梢越鈮嚎s，強(qiáng)制安裝（直接用設(shè)備管理器，顯卡更新驅(qū)動，解壓cuda11.8的安裝文件，找到里面的顯卡驅(qū)動文件夾）。直接用cuda的安裝會提示，因?yàn)橛行掳姹景惭b不成功。 - 驗(yàn)證cuda的版本是不是11.8
nvidia-smi
如果這個版本不是11.8。就是上面說的顯卡驅(qū)動的問題。需要強(qiáng)制安裝。
nvcc-V
這兩個都顯示cuda是11.8。就表示安裝成功
5.安裝pytorch
官網(wǎng)選擇cuda版本安裝。強(qiáng)烈不建議用conda，出現(xiàn)各種莫名其妙錯誤。安裝了一整天，不成功。
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
6.驗(yàn)證pytorch的版本和cuda是否可用文章來源：http://www.zghlxwxcb.cn/news/detail-470698.html
import torch
torch.cuda.is_available()
顯示gpu。就表示cuda可以用了。pytorch gpu版安裝成功。文章來源地址http://www.zghlxwxcb.cn/news/detail-470698.html
到了這里，關(guān)于2023最新pytorch2.0 gpu cuda安裝教程的文章就介紹完了。如果您還想了解更多內(nèi)容，請?jiān)谟疑辖撬阉鱐OY模板網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關(guān)文章，希望大家以后多多支持TOY模板網(wǎng)！
