J’ai eu l’erreur suivante :
NVIDIA GeForce RTX 3060 Laptop GPU with CUDA capability sm_86 is not compatible with the current PyTorch installation. The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_70. If you want to use the NVIDIA GeForce RTX 3060 Laptop GPU GPU with PyTorch, please check the instructions at https://pytorch.org/get-started/locally/
Pour fixer le problème j’ai fait:
pip3 install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html
Voici mon installation :
$ nvidia-smi Sat Mar 12 13:00:45 2022 +-----------------------------------------------------------------------------+ | NVIDIA-SMI 470.103.01 Driver Version: 470.103.01 CUDA Version: 11.4 | |-------------------------------+----------------------+----------------------+ | GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC | | Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. | | | | MIG M. | |===============================+======================+======================| | 0 NVIDIA GeForce ... Off | 00000000:01:00.0 On | N/A | | N/A 42C P0 27W / N/A | 1053MiB / 5946MiB | 0% Default | | | | N/A | +-------------------------------+----------------------+----------------------+ +-----------------------------------------------------------------------------+ | Processes: | | GPU GI CI PID Type Process name GPU Memory | | ID ID Usage | |=============================================================================| | 0 N/A N/A 1616 G /usr/lib/xorg/Xorg 608MiB | | 0 N/A N/A 3460 G budgie-wm 33MiB | | 0 N/A N/A 3959 G ...AAAAAAAA== --shared-files 2MiB | | 0 N/A N/A 4006 G ...AAAAAAAAA= --shared-files 60MiB | | 0 N/A N/A 4221 G ...AAAAAAAAA= --shared-files 42MiB | | 0 N/A N/A 4269 G ...AAAAAAAAA= --shared-files 15MiB | | 0 N/A N/A 4584 G ...oud-3.4.2-x86_64.AppImage 2MiB | | 0 N/A N/A 97021 G /usr/lib/firefox/firefox 148MiB | | 0 N/A N/A 99523 G ...rbird/192/thunderbird-bin 96MiB | | 0 N/A N/A 99893 G /usr/lib/firefox/firefox 2MiB | | 0 N/A N/A 100617 G ...826642640911437010,131072 33MiB | +-----------------------------------------------------------------------------+\