WSL CUDA安装
WSL CUDA安装
wsl的pci列表:
lspci
如果报错请安装pciutils
sudo apt install pciutils
安装cuda compiler tool
sudo apt install nvidia-cuda-toolkit
列出nVidia设备,输入:
nvidia-smi
CUDA Toolkit 12.4 Downloads | NVIDIA Developer
wget https://developer.download.nvidia.com/compute/cuda/12.4.0/local_installers/cuda_12.4.0_550.54.14_linux.runsudo sh cuda_12.4.0_550.54.14_linux.run
检查cuda是否安装成功:nvcc -V
如果没有这个命令的话,需要配置。
vim ~/.bashrc
在文件末尾添加(这里的/usr/local/cuda-11.3
替换成你的cuda安装路径,一般都在/usr/local
路径下)
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.3/lib64
export PATH=$PATH:/usr/local/cuda-11.3/bin
# export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-11.3
export CUDA_HOME=/usr/local/cuda-11.3
然后source ~/.bashrc
上述环境变量中,我看很多教程都是按照第三行配置的CUDA_HOME,但是后续我在使用deepspeed的时候会报错找不到/usr/local/cuda-11.3:/usr/local/cuda-11.3/bin/nvcc
,按照第四行配置CUDA_HOME就不会有这个问题。
再执行nvcc -V
就会显示出你刚刚安装的CUDA版本
评论
匿名评论
隐私政策
你无需删除空行,直接评论以获取最佳展示效果