ubuntu + cuda+cudnn+tensorflow-gpu+docker+tensorflow-serving

xiaoxiao2021-07-04  260

sudo dpkg -i cuda-repo-ubuntu1604-9-0-local_9.0.176-1_amd64.deb sudo apt-key add /var/cuda-repo-<version>/7fa2af80.pub sudo apt-get update sudo apt-get install cuda gredit ~/.bashrc export CUAD_HOME=/usr/local/cuda-9.0 export LD_LIBRARY=${CUDA_HOME}/lib64 export PATH=${PATH}:${CUDA_HOME}/bin source 1 sudo tar -xzvf cudnn-8.0-linux-x64-v5.1.tgz 2 sudo cp cuda/include/cudnn.h /usr/local/cuda/include 3 sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64 4 sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

sudo rm -rf /usr/local/cuda/include/cudnn.h

sudo rm -rf /usr/local/cuda/lib64/libcudnn*

cuda 版本 cat /usr/local/cuda/version.txt

cudnn 版本 cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

转载请注明原文地址: https://www.6miu.com/read-4821254.html

最新回复(0)