1.设置环境变量

将下面代码放入~/.bashrc文件并保存。

export@H_404_6@ LD_LIBRARY_PATH="$LD_LIBRARY_PATH@H_404_6@:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64"@H_404_6@
export@H_404_6@ CUDA_HOME=/usr/local/cuda

执行：

source@H_404_6@ ~/.bashrc

2.安装具有GPU功能的tensorflow

pip install@H_404_6@ tensorflow-gpu