深度學習環境配置

2021-10-08 15:17:33 字數 1061 閱讀 3159

參考:tensorrt7.0的安裝配置

匯入路徑export ld_library_path=$ld_library_path:/home/***/tensorrt-7.0.x/lib

更新 source ~/.bashrc

可以選擇在anaconda下新建tensorrt虛擬環境,本次選擇新建python3.5的tensorflow-gpu1.15環境

之後執行pip install pycuda

開啟解壓後的tensorrt資料夾下的python資料夾可以發現一共有

tensorrt-7.0.0.11-cp27-none-linux_x86_64.whl

tensorrt-7.0.0.11-cp34-none-linux_x86_64.whl

tensorrt-7.0.0.11-cp35-none-linux_x86_64.whl

tensorrt-7.0.0.11-cp36-none-linux_x86_64.whl

tensorrt-7.0.0.11-cp37-none-linux_x86_64.whl

根據當前的python版本進行選擇,本次選擇tensorrt-7.0.0.11-cp35-none-linux_x86_64.whl

執行

pip install tensorrt-

7.0.0

.11-cp35-none-linux_x86_64.whl

將目錄切換到tensorrt的uff目錄下執行

pip install uff-

0.6.5

-py2.py3-none-

any.whl

將目錄切換到tensorrt的graphsurgeon下執行

pip install graphsurgeon-

0.4.1

-py2.py3-none-

any.whl

安裝完成,import tensorflow;import tensorrt ;import uff驗證安裝是否成功

深度學習環境配置

ubuntu 16.04 cuda 8.0 cudnn 6.0 matlab r2017a tensorflow 1.4 anaconda3 本文用於記錄深度學習的各種環境配置,中間踩了不少的坑,各版本需要一一對應,故寫下本文以備不妨之需。cuda 8 最好對應 cudnn 6,tensorflow...

深度學習主機環境配置

1.安裝ubuntu16.04 2.更新軟體 1 輸入命令sudo apt update 2 輸入命令sudo apt upgrade 3.安裝1080ti顯示卡驅動 1 系統設定 軟體更新 附加驅動 選擇nvidia384 2 輸入命令 nvidia smi 可以看到具體的驅動資訊,如圖所示 4....

anaconda配置深度學習環境

1.cd到anaconda安裝包目錄下,安裝anaconda bash anaconda3 5.0.1 linux x86 64.sh2.按enter瀏覽完協議以後,輸入yes同意協議 注意再選擇安裝路徑的時候,按enter即可安裝在預設目錄下,不要再輸入yes 否則就安裝在yes目錄下了 t t ...