溫馨提示×

您好，登錄后才能下訂單哦！

密碼登錄×
登錄注冊(cè)×
其他方式登錄
點(diǎn)擊 登錄注冊(cè) 即表示同意《億速云用戶服務(wù)條款》

tensorflow 環(huán)境變量設(shè)置方式

發(fā)布時(shí)間：2020-10-24 08:31:33 來源：腳本之家 閱讀：276 作者：星辰大漠 欄目：開發(fā)技術(shù)

安裝TensorFlow后，在Python中輸入 import tensorflow as tf 時(shí) 提示一下類似錯(cuò)誤

ImportError: libcusolver.so.*.0: cannot open shared object file: No such file or directory

libcusolver.so..0 ( 代表某一版本)

解決辦法：

1. 輸入下面命令，查找libcusolver.so 相關(guān)文件的目錄

locate libcusolver.so.9

tensorflow 環(huán)境變量設(shè)置方式

2. 輸入下面命令 ， 將缺失文件的目錄添加到環(huán)境變量中。

export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda-9.0/lib64"

export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda-8.0/lib64"

輸入python , import tf 如果依然缺失文件，重復(fù) 1. 2.

然后

locate libcudnn.so.7

tensorflow 環(huán)境變量設(shè)置方式

export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/opt/cudnn/cudnn-v7.0.5-for-cuda8.0/lib64"

export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/opt/cudnn/cudnn-v5.1/lib64"

最終，成功import tensorflow

tensorflow 環(huán)境變量設(shè)置方式

如果需要長(zhǎng)期保持環(huán)境變量有效，要在 .bashrc文件中添加環(huán)境變量，如下

# added by Anaconda3 installer
export PATH="/home/yourname/anaconda3/bin:$PATH" 
export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda-8.0/lib64"
export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/opt/cudnn/cudnn-v5.1/lib64"

保存后激活 bashrc文件

source ~/.bashrc

以上這篇tensorflow 環(huán)境變量設(shè)置方式就是小編分享給大家的全部?jī)?nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持億速云。

向AI問一下細(xì)節(jié)

免責(zé)聲明：本站發(fā)布的內(nèi)容（圖片、視頻和文字）以原創(chuàng)、轉(zhuǎn)載和分享為主，文章觀點(diǎn)不代表本網(wǎng)站立場(chǎng)，如果涉及侵權(quán)請(qǐng)聯(lián)系站長(zhǎng)郵箱：is@yisu.com進(jìn)行舉報(bào)，并提供相關(guān)證據(jù)，一經(jīng)查實(shí)，將立刻刪除涉嫌侵權(quán)內(nèi)容。

AI
助
手