[深度学习]OSError: CUDA

• 人工智能

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

本文来自网络，不代表协通编程立场，如若转载，请注明出处：https://www.net2asp.com/fb9ed29e91.html

赞 (0)
« 上一篇 1天前
下一篇 » 1天前

相关推荐