ubuntu22.04에서 RTX 4090 그래픽 카드를 교체한 후 드라이버와 파이토치 레코드를 설치합니다.

전제조건 환경:

그래픽 카드를 변경하기 전(이전에는 2080ti 사용 중) 우분투 그래픽 인터페이스의 "추가 드라이버"에 nvidia 드라이버가 설치되어 있었고, 아나콘다 환경에서는 파이토치가 정상적으로 작동하고 있었습니다.

물론 가장 먼저 위대한 Hushubao를 제시하는 것입니다.

1. 공식 웹사이트에서 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