当前位置:   article > 正文

cuda&cudnn环境变量配置_cudnn环境变量设置

cudnn环境变量设置

cuda&cudnn配置备忘

cuda:

打开:
/etc/profile
  • 1
  • 2

写入

export PATH=$PATH:/usr/local/cuda/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
  • 1
  • 2

执行:

source /etc/profile
  • 1

测试:

nvcc -V
  • 1
cat /usr/local/cuda/version.json
  • 1

cudnn:

sudo cp include/* /usr/local/cuda-11.7/include && \
sudo cp -P lib/* /usr/local/cuda-11.7/lib64 && \
sudo chmod a+r /usr/local/cuda-11.7/include/cudnn*.h /usr/local/cuda-11.7/lib64/libcudnn*
  • 1
  • 2
  • 3

测试:

cat /usr/local/cuda/include/cudnn_version.h
  • 1
声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/我家小花儿/article/detail/151256
推荐阅读
相关标签
  

闽ICP备14008679号