Hi everybody,
I have problems to use my GPUs with torch in Ubuntu 20.04 and my GeForce RTX 3090 and CUDA 11.2.
-
I installed according to Start Locally | PyTorch and adapt the cudatoolkit from 11.1 to 11.2: conda install pytorch torchvision torchaudio cudatoolkit=11.2 -c pytorch -c conda-forge
-
print(torch.version)
1.8.0 -
I installed cuda by: https://developer.nvidia.com/cuda-downloads?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=2004&target_type=deblocal
-
nvidia-smi says: | NVIDIA-SMI 460.32.03 Driver Version: 460.32.03 CUDA Version: 11.2
-
nvcc --version says: nvcc --version
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2021 NVIDIA Corporation
Built on Sun_Feb_14_21:12:58_PST_2021
Cuda compilation tools, release 11.2, V11.2.152
Build cuda_11.2.r11.2/compiler.29618528_0 -
torch.cuda.is_available()
False
Does anybody have a tip or an idea how I can use my GPU?
Thanks and best regards,
Ilja