[深度学习]OSError: CUDA_HOME environment variable is not set. Please s

[复制链接]
查看598 | 回复0 | 2023-8-23 11:44:17 | 显示全部楼层 |阅读模式
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:
  1. import os
  2. print(os.environ.get('CUDA_PATH'))
复制代码
OUTPUT:
  1. C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1
复制代码
来源:https://blog.csdn.net/weixin_41519550/article/details/130050863
免责声明:如果侵犯了您的权益,请联系站长,我们会及时删除侵权内容,谢谢合作!
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则