目录
当前位置: 首页 > 文档资料 > Caffe 学习笔记 >

Tensorflow

优质
小牛编辑
136浏览
2023-12-01

TF install

install cuda

cuda 有很多版本,要仔细看清自己需要什么样的版本:

TF cuda 版本对照

error

import tensorflow出现:ImportError: libcublas.so.9.0: cannot open shared object file: No such file or directory

尝试使用下面方法解决:

在 PATH中 加入 cuda

export PATH=/usr/local/cuda/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
export CUDA_HOME=/usr/local/cuda
或者
export LD_LIBRARY_PATH="/usr/local/cuda-9.0/lib64:/usr/local/cuda-9.0/extras/CUPTI/lib64"

ldconfig

sudo ldconfig /usr/local/cuda-9.0/lib64

example