在python3.8的环境下安装
pip install nvidia-pyindex
pip install nvidia-tensorflow
报错“cudaGetDevice() failed. Status: CUDA driver version is insufficient for CUDA runtime version”
解决办法很简单,就是下载低版本的轮子,手动安装,
URL:https://developer.download.nvidia.cn/compute/redist/nvidia-tensorflow/
eg:nvidia_tensorflow-1.15.5+nv21.02-cp38-cp38-linux_x86_64.whl
其他包手动调整版本即可
NVIDIA A10/A100显卡安装tf1.15环境
最新推荐文章于 2024-05-17 14:39:55 发布