[深度学习]OSError: CUDA_HOME environment variable is not set. Please set it to your CUDA install root.
admin 阅读： 2024-03-26
| 后台-插件-广告管理-内容页头部广告（手机） |
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'))- 1
- 2
OUTPUT:
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1- 1
声明
1.本站遵循行业规范，任何转载的稿件都会明确标注作者和来源；2.本站的原创文章，请转载时务必注明文章作者和来源，不尊重原创的行为我们将追究责任；3.作者投稿可能会经我们编辑修改或补充。
在线投稿：投稿 站长QQ：1888636
| 后台-插件-广告管理-内容页尾部广告（手机） |
