ubuntu22.04 で RTX 4090 グラフィックスカードを交換した後、ドライバーと pytorch レコードをインストールします

前提環境:

グラフィックカードを変更する前（以前は2080tiを使用していました）、ubuntuグラフィカルインターフェイスの「追加ドライバー」にnvidiaドライバーがインストールされており、anaconda環境のpytorchは正常に動作しています。

もちろん、何よりもまず、偉大な胡舟宝を披露することです。

1. 公式 Web サイトから 4090 グラフィックスカードドライバーをダウンロードします。

注: 「追加ドライバー」に nvida ドライバーがインストールされていることを確認してください。そうでない場合は、手動で nouveau を無効にしてください。

2. GUIを閉じます。

sudo systemctl set-default multi-user.target
sudo reboot

グラフィカルインターフェイスを開くための対応するコマンドは次のとおりです (一時的には開きません)

sudo systemctl set-default graphical.target
sudo reboot

3. tty ログイン後、新しいドライバーをインストールします

# 切换到驱动目录
cd /home/user/download/
# 权限
sudo chmod a+x NVIDIA-Linux-x86_64-520.56.06.run
# 安装
sudo ./NVIDIA-Linux-x86_64-520.56.06.run

インストールプロセス中に、32 ビットをインストールするオプションがあります。 [いいえ]を選択します。残りはずっと「はい」です。

4. GUIを開いて再起動します。

sudo systemctl set-default graphical.target
sudo reboot

インストール後の初回起動には時間がかかりますので、しばらくお待ちください。

4090 グラフィックスカードを交換すると、cuda10.2 に基づいてコンパイルされた pytorch がサポートされなくなることが確認されました。

cuda11.6でコンパイルしたpytorchは普通に使えます。

conda install pytorch torchvision torchaudio cudatoolkit=11.6 -c pytorch -c conda-forge

ubuntu22.04 で RTX 4090 グラフィックス カードを交換した後、ドライバーと pytorch レコードをインストールします