tensorflow-GPUセットアップスクリプト

関連ダウンロード:

https://pan.baidu.com/s/1EkmBzPtprn-aiE0ogVyHpQ

 

#!/ binに/ bashの
インストールスクリプトの#tensorflow-GPU版


#ドライバインストール
のUbuntu-ドライバのデバイスは、
NVIDIA-APT-GET-435ドライバーのインストールはsudo 
のsudoの再起動

#10.1インストールCUDAの
はchmod + X-cuda_10.1.105_418.39_linux.runの
sudoを。 /cuda_10.1.105_418.39_linux.runの

#cudnn取り付け
タールcudnn xzf-xのx64のLinux-10.1-v7.6.4.38.tgz 
CP CUDA /含める/ * /usr/local/cuda-10.1/include/のsudo 
のsudo CP CUDA / lib64に/ * /usr/local/cuda-10.1/lib64/ 

#環境変数の
エコー'輸出PATHは=は/ usr / local / CUDA-10.1 / binに:PATH $' >> / etc / profileを
ソースには/ etc / profile 
にldconfigを

#マウントTensorRT 
タール6.0.1.5.Ubuntu TensorRT-18.04.x86_64-gnu.cuda-10.1.cudnn7.6.tar.gz xzvf 

CD-TensorRT 6.0.1.5 /パイソン
sudoのPIP3 tensorrt-6.0.1.5-cp36-なし- linux_x86_64.whlインストール

#安装UFF 
CD ../uff 
sudoのPIP3インストールUFF-0.6.5-py2.py3-なし- any.whl 

#安装graphsurgeon 
のCD ../graphsurgeonを
sudoのPIP3インストールgraphsurgeon-0.4.1-py2.py3-なし- any.whl 

#配置环境变量
CD ../ 
のsudoのsh -c「エコー$(PWD)/TensorRT-6.0.1.5/lib>は/ etc / LD。 so.conf.d / TensorRT.conf」
にsudo ldconfigを

  

おすすめ

転載: www.cnblogs.com/springwind2006/p/12194767.html