Torch not detecting nVidia GPU #107

vladmandic started this conversation in General

vladmandic
Apr 13, 2023
Maintainer

If you get Torch repoorts CUDA not available and you do have nVidia GPU then you need to install nVidia CUDA Toolkit:

Install CUDA Toolkit from https://developer.nvidia.com/cuda-downloads
Run nvidia-smi to confirm CUDA is installed and GPU is visible
Uninstall Torch so it can be reinstalled with CUDA support

pip uninstall torch torchvision torchaudio

Replies: 0 comments

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment