目前网络上的几乎所有环境变量添加都是错的,在/usr/local/cuda-11.8下根本没有lib64这个文件夹!CUDNN通过deb文件安装后,头文件和库文件也不在cuda-11.8的根目录下!害我跑代码各种报错,于是个人决定亲自搜索一番
以下环境变量路径均为个人搜索,并自定义设置,完全可用
向用户根目录下的.bashrc文件中添加如下内容
#>>> cuda PATH >>>
export CUDA_HOME=:/usr/local/cuda-11.8
export PATH=“$CUDA_HOME:$CUDA_HOME/bin:$PATH”
export CUDA_LIB_PATH=“$CUDA_HOME/targets/x86_64-linux/lib”
export LD_LIBRARY_PATH=“$CUDA_LIB_PATH:$CUDA_LIB_PATH/stubs:$LD_LIBRARY_PATH”
#>>> cuDNN PATH>>>
export CUDNN_INCLUDE_PATH=“/usr/include”
export PATH=“$CUDNN_INCLUDE_PATH:$CUDNN_INCLUDE_PATH/x86_64-linux-gnu:$PATH”
export CUDNN_LIB_PATH=“/usr/lib/x86_64-linux-gnu”
export LD_LIBRARY_PATH=“$CUDNN_LIB_PATH:$LD_LIBRARY_PATH”

基于WSL2的Linux子系统是真的好用啊,上文内容可以完全在Windows资源管理器中操作,炼丹也比Windows快,多卡可无障碍使用NCCL,香

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐