[深度学习]OSError: CUDA_HOME environment variable is not set. Please set it to your CUDA install root.
【代码】[深度学习]OSError: 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 Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1
加入「COC·上海城市开发者社区」，成就更好的自己！
更多推荐
2
0- 0
已为社区贡献1条内容
所有评论(0)