1.设置环境变量
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
@H_301_20@2.安装具有GPU功能的tensorflow
pip install tensorflow-gpu
原文链接:https://www.f2er.com/ubuntu/352558.html