1.服务器重启后torch不能用cuda,使用tensorflow的时候发现报错
cuda_driver.cc:406] failed call to cuInit: CUDA_ERROR_UNKNOWN的错误
安装sudo apt-get install nvidia-modprobe
2.缺少CUPTI
export LD_LIBRART_PATH=/usr/local/cuda/extras/CUPTI/lib64
本文介绍了服务器重启后遇到的torch无法使用cuda及tensorflow报错的问题,并提供了具体的解决方案,包括安装nvidia-modprobe和设置CUPTI环境变量。
1.服务器重启后torch不能用cuda,使用tensorflow的时候发现报错
cuda_driver.cc:406] failed call to cuInit: CUDA_ERROR_UNKNOWN的错误
安装sudo apt-get install nvidia-modprobe
2.缺少CUPTI
export LD_LIBRART_PATH=/usr/local/cuda/extras/CUPTI/lib64
您可能感兴趣的与本文相关的镜像
PyTorch 2.8
PyTorch 是一个开源的 Python 机器学习库,基于 Torch 库,底层由 C++ 实现,应用于人工智能领域,如计算机视觉和自然语言处理
4万+
1万+

被折叠的 条评论
为什么被折叠?