Caffe2 新增CUDA路徑

2022-08-22 07:54:09 字數 652 閱讀 6032

在使用eclipse時候,仍然出現 libcuda.so 找不到的情況.....

原因:cuda環境沒有設定到系統環境中。

方案:設定cuda path:

在bashrc中新增路徑,只對當前使用者產生影響。

sudo gedit ~/.bashrc
在最後行新增內容:

export path=/usr/local/cuda/bin:$path  

export ld_library_path=/usr/local/cuda/lib64:$ld_library_path

儲存後, 執行下列命令, 使環境變數立即生效,可以暫時不進行重啟

source ~/.bashrc
同時需要新增lib庫路徑: 在 /etc/ld.so.conf.d/加入檔案 cuda.conf,

sudo gedit /etc/ld.so.conf.d/cuda.conf
內容如下

/usr/local/cuda/lib64 

/lib

/usr/lib

/usr/lib32

儲存後,執行下列命令使之立刻生效

sudo ldconfig -v

結果,匯入成功。

caffe2 安裝與介紹

一早發現caffe2的較成熟的release版發布了 the first production ready release 那麼深度學習平台在之後一段時間也是會出現其與tensorflow相互競爭的局面。從開啟這個caffe2的官網就會發現,有了facebook的支援,連介面也好看多了。不過再仔細看...

Caffe2的安裝詳述

caffe2的安裝過程區分ubuntu的不同版本,在ubuntu 14.04和 ubuntu 16.04下的相關命令有所不同,最重要的是caffe2主要支援python2.7版本,以下python和pip命令預設情況下都是指python2.7版本,經過筆者測試,caffe2版本暫時不支援python...

caffe 轉caffe2模型出現的問題

剛開始接觸caffe,用nvidia tx2跑了ssd模型,達到8 9幀。把caffe模型轉caffe2模型時,轉到reshape層時出現錯誤 valueerror unknown argument type key values value 發現是protobuffer 版本問題,參考 解壓,編譯...