Search
 
SCRIPT & CODE EXAMPLE
 
CODE EXAMPLE FOR SHELL

cuda_home environment variable is not set. please set it to your cuda install root.

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 FilesNVIDIA GPU Computing ToolkitCUDAv10.1
 
PREVIOUS NEXT
Tagged: #environment #variable #set #cuda #install
ADD COMMENT
Topic
Name
9+1 =