cudaバージョンに一貫性がありません。システムのNVIDIAドライバーが古すぎます

遭遇問題

システムのNVIDIAドライバーが古すぎます(バージョン9010が見つかりました)。

のURLから新しいバージョンをダウンロードしてインストールすることにより、GPUドライバーを更新してください。http

//www.nvidia.com/Download/index.aspxまたは、https://pytorch.orgにアクセスして、次のバージョンのPyTorchをインストールします。お使いのバージョン
のCUDAドライバーでコンパイルされています
(mmlab)root @ bda240ba6e9c:/ home / 41228 / zlk / VQA_DL#cat / proc / driver / nvidia / version
NVRMバージョン:NVIDIA UNIXx86_64カーネルモジュール387.26Thu Nov 2 21:20:16 PDT 2017
GCCバージョン:gccバージョン4.8 .5 20150623(Red Hat 4.8.5-16)(GCC)

解決

pytorchのトーチバージョンとtochvisionバージョンの不整合に加えて、cudaバージョンも不整合である可能性があります。上記のバージョンが同じであっても、nvidiaドライバーのバージョンが古すぎるか新すぎるため、エラーが発生します。

  • GPUモデルを表示
lspci | grep -i nvidia
  • NVIDIAドライバーのバージョン
    を確認します。440、364など対応するバージョンは、公式Webサイトにアクセスして、対応するcudaバージョン10または9を確認できます。
sudo dpkg --list | grep nvidia-*
或者
cat /proc/driver/nvidia/version

おすすめ

転載: blog.csdn.net/u012457196/article/details/108336983