添加环境变量:
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
export PATH=$PATH:/usr/local/cuda/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda
查看cudnn版本:
cat /usr/local/cuda/include/cudnn_version.h| grep CUDNN_MAJOR -A 2
添加环境变量:
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
export PATH=$PATH:/usr/local/cuda/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda
查看cudnn版本:
cat /usr/local/cuda/include/cudnn_version.h| grep CUDNN_MAJOR -A 2