torch.cuda.is_available()返回false
本人誤打誤撞似乎解決了這個問題,記錄一下,如果同學沒有其他的解決方案,可以參考本文。
在一篇博客上看到,說問題是清華源有誤,下載下來的就是cpu版本,本人也嘗試了很多方法,最后使用離線下載解決,簡單寫一下步驟:

步驟一:下載
下載鏈接:(可用迅雷下載)
https://download.pytorch.org/whl/torch_stable.html??
cu102:? ?cuda版本10.2
torch-1.7.0:? ?torch版本1.7.0
cp37:? ? python版本3.7(后面的cp37m不知道是啥)
linux:? ? 操作系統(tǒng)linux
x86_64:? 同時兼容32和64位系統(tǒng)


步驟二:安裝
將上面的包放入下面的位置(其他位置不知道行不行),在上面的地址欄輸入cmd,輸入命令,中間第三個參數(shù)是自己下載的包名,后面的參數(shù)是豆瓣源,不知道有沒有用,都試試
pip install "torch-1.6.0-cp36-cp36m-win_amd64.whl" -i https://pypi.douban.com/simple
pip install "torchvision-0.7.0+cu101-cp36-cp36m-win_amd64.whl" -i https://pypi.douban.com/simple?


步驟三:驗證

注:cuda10.2?和 cudnn安裝包下載(可忽略)
鏈接:
https://blog.csdn.net/weixin_55775980/article/details/119768352
標簽: