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

时间:2024-02-15 15:56:15

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

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

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,直接运行飞浆提示错误:

>>> 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,将找到的目录加入环境变量即可:

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

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