RTX 3090 の計算能力は 8.6 で、pytorch のバージョンは 11.1 以降である必要があります。そうでない場合、コードは GPU で実行できず、対応する cuda バージョンも更新する必要があります。
1、元の cuda をアンインストールする
sudo /usr/local/cuda-10.0/bin/uninstall_cuda_10.0.pl
sudo rm -rf /usr/local/cuda-10.0次に、cudaをインストールします
cuda11.1 Web サイト: CUDA ツールキット 11.1.0 | NVIDIA 開発者
選択後、以下のリンクを Linux ターミナルにコピーします。
sudo sh cuda_11.1.0_455.23.05_linux.run
インストールプロセス中に、次のインターフェイスがポップアップ表示されます
グラフィックス カード ドライバーは以前にインストールされているため、ここでドライバーを選択し、[Enter] をクリックして X 番号を削除し、[インストール] を選択して Enter キーを押して通常どおりインストールします。次の行
エクスポート LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
エクスポート PATH=$PATH:/usr/local/cuda/bin
エクスポート CUDA_HOME=$CUDA_HOME:/usr/local/cuda
保存後、終了してsource ~/.bashrcを入力してアクティブ化します。
ターミナルに nvcc -V と入力して、cuda が正常にインストールされているかどうかを確認します。
cuda11.1バージョンに対応したpytorchを再ダウンロードする
conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch-lts -c nvidia