在训练模型时使用print(tf.test.is_gpu_available())
显示True,但是在训练时仍然会显示
Couldn't open CUDA library libcupti.so 。
是因为没有为libcupti添加环境变量,其实这一点在官方文档中已经指出了,再次证明查阅官方文档的重要性,不慌,只需要先
cd ~
vi .bashrc
之后添加
export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64"
export CUDA_HOME=/usr/local/cuda
然后
source .bashrc
重新导入环境变量即可
附上一张成功的截图