在遠程服務器安裝pytorch,根據(jù)官網(wǎng)命令進行安裝,但在完成之后,顯示GPU不可用,故記錄此大坑。
一、根據(jù)官網(wǎng)進行安裝
?安裝的很快,但是?。“惭b結(jié)束之后,輸入以下代碼進行安裝驗證卻顯示沒有成功安裝?。?/p>
import torch # 如果pytorch安裝成功即可導入
print(torch.cuda.is_available()) # 查看CUDA是否可用
print(torch.cuda.device_count()) # 查看可用的CUDA數(shù)量
print(torch.version.cuda) # 查看CUDA的版本號
所以這表明安裝大失??!
但是不死心的我又輸入以下語句來檢查torch
conda list
這表明我們安裝的只是CPU版本!!
二、解決方法
因為這個大坑,可以說浪費了一整個晚上的時間,在借鑒了很多博客之后,下面這個博主的博客可以說是一個超級好的解決方法。
pytorch無法使用cuda
按照這個博客的引導,使用pip成功安裝GPU版本的torch
文章來源:http://www.zghlxwxcb.cn/news/detail-503493.html
特此記錄文章來源地址http://www.zghlxwxcb.cn/news/detail-503493.html
到了這里,關于GPU版pytorch安裝成功卻無法使用cuda的文章就介紹完了。如果您還想了解更多內(nèi)容,請在右上角搜索TOY模板網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關文章,希望大家以后多多支持TOY模板網(wǎng)!