- 問題
使用pytorch官網的安裝命令安裝的pytorch可能只是支持CPU版本的

本文通過下載安裝包的方式安裝支持GPU的pytorch。 - 下載安裝包
在pytorch開源官網https:download.pytorch.org/whl/torch_stable.html里找到對應的安裝包,注意需要下載torch和torchvision兩個安裝包


- 安裝
下載后上傳到機器上,在目錄下執行使用pip install安裝即可

在執行命令后加上清華源,不加可能因網絡超時而導致安裝失敗 - 驗證
pip list可以看到torch和torchvision安裝的為gpu版本





