リモートサーバーに pytorch をインストールし、公式 Web サイトのコマンドに従いますが、完了後に GPU が使用できないと表示されるため、この落とし穴を記録します。
1.公式サイトに従ってインストールする
とにかくインストールが早い!!インストール完了後、インストール確認のために次のコードを入力しましたが、インストールが成功しなかったことが表示されました。!
import torch # 如果pytorch安装成功即可导入
print(torch.cuda.is_available()) # 查看CUDA是否可用
print(torch.cuda.device_count()) # 查看可用的CUDA数量
print(torch.version.cuda) # 查看CUDA的版本号
つまり、これはインストールに大きな失敗があることを示しています。
でも諦めず、トーチを確認するために次のステートメントを入力しました
conda list
これは、CPU バージョンのみをインストールしていることを示しています。!
2. 解決策
この大きな落とし穴のせいで、一晩が無駄になったと言えますが、多くのブログから学んだ結果、次のブロガーのブログは非常に優れた解決策であると言えます。
このブログのガイダンスに従って、pip を使用して GPU バージョンの torch を正常にインストールします。
ここに記録されています