paddlepaddle 2.6版本在WSL2环境中如何使用NVIDIA显卡运行神经网络

kampoo2024-02-15 8:49

paddlepaddle 2.6版本发布后，官网上可以使用NVIDIA cuda 12.x进行机器学习了，训练神经网络的效率大为提升。因为是在wsl2环境中安装，不是纯正的linux环境，其中一些小问题需要注意。

使用conda 安装飞浆，wsl2中安装了cuda 12.x，跟飞浆2.6兼容，按照官网指令即可：

bash 复制代码

conda create -n pp2cuda python=3.11
conda activate pp2cuda
conda install paddlepaddle-gpu==2.6.0 cudatoolkit=11.7 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/Paddle/ -c conda-forge

启动python，直接运行飞浆提示错误：

python 复制代码

>>> paddle.utils.run_check()
# The third-party dynamic library (libcuda.so) that Paddle depends on 
# is not configured correctly. (error code is libcuda.so: cannot open 
# shared object file: No such file or directory)

错误提示 libcuda.so找不到。在 /usr目录下查找：find /usr -name libcuda.so，将找到的目录加入环境变量即可：

bash 复制代码

# paddlepaddle 2.6.0 cuda
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/targets/x86_64-linux/lib/stubs/

再次测试飞浆，paddlepaddle-gpu可以正常通过了。