CUDA unavailable on GPU platform

Issue descriptionIssue description

CUDA is not available on the GPU platform.

SolutionSolution

Run these commands:

1. sudo apt update -y
2. sudo apt upgrade -y
3. pip3 install torch

Check whether CUDA is available by calling torch.cuda.is_available().

You may also consider using DataSphere that leverages GPU configurations. Every service project contains multiple Docker images with a specific software stack. You can still build a custom image for your project, if required.

To check the version of these components, use the following commands:

• !nvcc --version for CUDA.
• !cat /usr/include/cudnn_version.h for cuDNN.

You might also want to check out:

• Info on NVIDIA by running !nvidia-smi.
• CUDA availability using torch by calling torch.cuda.is_available().