每次換個環(huán)境又要安裝pytorch,最開始安裝總是出錯,寫下不踩坑安裝全過程
1、首先看一下cuda的版本
運行命令?nvidia-smi
如下所示
?比方說我的是右上角顯示11.2,所以
2、去PyTorch 官網(wǎng)
?選擇Linux ,這里沒有我的cuda版本
?
?找過往版本Previous PyTorch Versions | PyTorch
?發(fā)現(xiàn)有11.3? 11.1版本和10.2版本,但是沒有11.2版本,查閱得知可以向下兼容安裝11.1版本的
?3、往下滑找到cuda對應(yīng)版本(11.1)版本的torch
?按照以下命令安裝
pip install torch==1.10.1+cu111 torchvision==0.11.2+cu111 torchaudio==0.10.1 -f https://download.pytorch.org/whl/cu111/torch_stable.html
等待……
安裝成功
4、檢查是否安裝成功
以及確定是不是gpu版本的
?輸入python回車
文章來源:http://www.zghlxwxcb.cn/news/detail-766325.html
返回true代表安裝gpu版本的torch成功文章來源地址http://www.zghlxwxcb.cn/news/detail-766325.html
到了這里,關(guān)于linux服務(wù)器安裝gpu版本的pytorch的文章就介紹完了。如果您還想了解更多內(nèi)容,請在右上角搜索TOY模板網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關(guān)文章,希望大家以后多多支持TOY模板網(wǎng)!