OSError: CUDA_HOME environment variable is not set

meder411 / PyTorch-EMDLoss

PyTorch 1.0 implementation of the approximate Earth Mover's Distance

136 stars 13 forks source link

Open icoz69 opened 5 years ago

icoz69 commented 5 years ago

hi, when i run the setup.py, came across this issus. how can i slove it?

Leong1230 commented 3 years ago

Just add the line in your .bashrc to configurate your cuda environment: export CUDA_HOME=/usr/local/cuda and source ~/.bashrc