從官網(wǎng)一鍵命令下載的pytorch是CPU版本的，我們需要手動(dòng)安裝GPU版本的
如圖，這樣是不能安裝gpu版本的。
解決方案
查找對(duì)應(yīng)版本
這里針對(duì)python3.9版本，在此網(wǎng)站https://github.com/pytorch/vision查找讀者python版本對(duì)應(yīng)的torch和torchvision。
下載對(duì)應(yīng)的whl文件
然后在這個(gè)網(wǎng)址https://download.pytorch.org/whl/torch/ 和 https://download.pytorch.org/whl/torchvision/里下載，我這里是對(duì)應(yīng)了這兩個(gè)。千萬要對(duì)應(yīng)，不然會(huì)報(bào)錯(cuò)。
使用pip本地安裝
先進(jìn)入要安裝的python環(huán)境，base或者虛擬環(huán)境，我這里在虛擬環(huán)境里裝。（虛擬環(huán)境是anaconda里創(chuàng)建的，不知道虛擬環(huán)境的搜索下別人的博客先學(xué)習(xí)一下）。上面下載完的兩個(gè)文件放在要安裝的環(huán)境根目錄中文章來源：http://www.zghlxwxcb.cn/news/detail-458954.html
pip install .\torch-1.13.0+cu116-cp39-cp39-win_amd64.whl
pip install .\torchvision-0.14.0+cu116-cp39-cp39-win_amd64.whl
如圖successfully就可以了。文章來源地址http://www.zghlxwxcb.cn/news/detail-458954.html
可能遇到的問題
numpy和pandas報(bào)錯(cuò)，uninstall后重新install就行
最后測(cè)試
import torch
print(torch.cuda.is_available())
//True
到了這里，關(guān)于2022-2023最新 pytorch安裝方法 GPU版本 python3.9 torch-1.13.0+cu116-cp39 torchvision-0.14.0 親自安裝可用！（其他版本也通用）的文章就介紹完了。如果您還想了解更多內(nèi)容，請(qǐng)?jiān)谟疑辖撬阉鱐OY模板網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關(guān)文章，希望大家以后多多支持TOY模板網(wǎng)！
