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
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章