Ubuntu16.04安装TensorFlow教程

前端之家收集整理的这篇文章主要介绍了Ubuntu16.04安装TensorFlow教程前端之家小编觉得挺不错的,现在分享给大家,也给大家做个参考。

1.设置环境变量

将下面代码放入~/.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
@H_301_20@2.安装具有GPU功能的tensorflow
pip install tensorflow-gpu
原文链接:https://www.f2er.com/ubuntu/352558.html

猜你在找的Ubuntu相关文章