I am facing the exact error. Following your advice I did the following:
- To install cuda 11 I used following command:
sudo sh cuda_11.4.0_470.42.01_linux.run --toolkit --silent --override
And I confirm to see the following when I use nvcc -V linux command:
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2021 NVIDIA Corporation
Built on Wed_Jun__2_19:15:15_PDT_2021
Cuda compilation tools, release 11.4, V11.4.48
Build cuda_11.4.r11.4/compiler.30033411_0
- To install a compatible torch for the installed cudatoolkit I used following pip command
pip3 install torch==1.9.0 -f https://download.pytorch.org/whl/cu114/torch_stable.html
However, I still get the above error! I am wondering if you might have any advice/comment?
Thanks