This note is for things with PyTorch. Personal notes only.
- Verify that your computer has a graphic card (NVIDIA):
lspci -nn | grep '[03'
- Check the CUDA version on Ubuntu:
nvidia-smi
. If have problems, check this note.
- Install the lates version: here.
For example, need to install corresponding versions:
torch==1.2.0
<- torchvision==0.4.0
<- Pillow<7.0.0
Another option (worked on XPS 15 7950):
torch==1.5.1
, torchvision==0.6.1
, pillow==7.2.0
(with nvcc --version
is 11.1
).Problem
The NVIDIA driver on your system is too old (found version 10010)
.From this website, problem is solved by
- Switching from
nvidia-driver-435
tonvidia-driver-440
.
- Restart the computer.
It works on Dell XPS 7950 whose GPU is NVIDIA GTX 1650.
🔅 RuntimeError: cuda runtime error (804) : forward compatibility was attempted on non supported HW at /pytorch/aten/src/THC/THCGeneral.cpp:47 (after update system including nvdia-cli, maybe): check this note.