“Variabel lingkungan cuda_home tidak diatur. Harap atur ke root instalasi CUDA Anda.” Kode Jawaban

Variabel lingkungan cuda_home tidak diatur. Harap atur ke root instalasi CUDA Anda.

Solution to above issue!

As cuda installed through anaconda is not the entire package. Please install cuda drivers manually from Nvidia Website[ https://developer.nvidia.com/cuda-downloads ]

After installation of drivers, pytorch would be able to access the cuda path. You can test the cuda path using below sample code. Problem resolved!!!

CHECK INSTALLATION:

import os
print(os.environ.get('CUDA_PATH'))
OUTPUT:

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1