Ubuntu上tensorflow设置环境变量

来源:互联网 发布:淘宝马云怎么赚钱 编辑:程序博客网 时间:2024/06/08 18:29

在ubuntu上使用import tensorflow 会出现环境变量错误的情况,此时需要添加环境变量。

1.先在terminal上打开环境变量文件

gedit ~/.bashrc

2.在bashrc文件后面添加环境变量

export LD_LIBRARY_PATH="LD_LIBRARY_PATH:/usr/local/cuda/lib64"export CUDA_HOME=/usr/local/cuda

此时tensorflow就可以正常使用

原创粉丝点击