????????每次從github上安裝項(xiàng)目對(duì)應(yīng)的庫(kù),利用requirements.txt安裝很容易出現(xiàn)版本不對(duì)應(yīng)的情況,尤其是將torch的gpu版本安裝成cpu。這里記錄一些查看版本的指令和離線(xiàn)安裝的方法,就不用每次百度啦?。ㄗⅲ浩渌麕?kù)的離線(xiàn)安裝也可以用同樣的方法,只需要去相應(yīng)的網(wǎng)站下載wheel即可)
1.首先查看torch版本(指令最好在終端一行一行地輸入)
activate pytorch #進(jìn)入環(huán)境
python
import torch
print(torch.__version__)#查看torch版本
print(torch.cuda.is_available())#查看cuda能用嗎
如果版本是cpu和輸出false,可以選擇卸載并重新安裝,這里用離線(xiàn)安裝的方式最穩(wěn):
pip uninstall torch #卸載之前安裝的torch
python -m pip debug --verbose #查看自己能安裝的wheel版本
?注意一下紅框部分
?2.然后下載離線(xiàn)資源?https://download.pytorch.org/whl/torch_stable.html
?第一個(gè)框cu表示gpu版本,102表示版本為10.2,第二個(gè)框表示torch,一般也同時(shí)安裝torchversion,往下拉選擇相同的版本就好,第三個(gè)框與上一步紅框相對(duì)應(yīng)。
3.離線(xiàn)安裝
cd C:\Users\1\Desktop #進(jìn)到你文件下載的地方
pip install torch-1.9.1+cu102-cp37-cp37m-win_amd64.whl #安裝,install后面是你下載的文件名加上后綴.whl
pip install torchvision-0.10.0-cp37-cp37m-win_amd64.whl #安裝torchversion
?4.然后用步驟一的指令檢驗(yàn)一下歐克了:文章來(lái)源:http://www.zghlxwxcb.cn/news/detail-540970.html
文章來(lái)源地址http://www.zghlxwxcb.cn/news/detail-540970.html
到了這里,關(guān)于卸載cpu版本的torch并離線(xiàn)安裝對(duì)應(yīng)的gpu版本的文章就介紹完了。如果您還想了解更多內(nèi)容,請(qǐng)?jiān)谟疑辖撬阉鱐OY模板網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關(guān)文章,希望大家以后多多支持TOY模板網(wǎng)!