tensorflow不能识别gpu
第0步:卸载protobufpip uninstall protobuf第1步:卸载tensorflowpip uninstall tensorflowpip uninstall tensorflow-gpu第2步:强制重新安装Tensorflow并支持GPUpip install --upgrade --force-reinstall tensorflow-gpu...
·
第0步:卸载protobuf
pip uninstall protobuf
第1步:卸载tensorflow
pip uninstall tensorflow
pip uninstall tensorflow-gpu
第2步:强制重新安装Tensorflow并支持GPU
pip install --upgrade --force-reinstall tensorflow-gpu
第3步:如果还没有,请设置CUDA_VISIBLE_DEVICES
所以对我来说有2个GPU就可以了
export CUDA_VISIBLE_DEVICES=0,1更多推荐
所有评论(0)