wsl安装深度学习基础环境
1.查看显卡cuda版本
打开NVIDIA显卡控制面板,查看系统信息
2.安装CUDA Toolkit
按照官网下载安装
打开官网:CUDA Toolkit Archive | NVIDIA Developer
选择要安装的版本12.6.0,要和自己电脑上驱动的cuda版本适配。
环境变量配置
sudo vim ~/.bashrc
将以下内容添加,如更换版本,注意更换目录名称
export PATH=/usr/local/cuda-12.6/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-12.6/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
更新环境变量
source ~/.bashrc
测试安装是否成功,显示版本即成功。
(base) jason@Jason-tx:~$ nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2024 NVIDIA Corporation
Built on Fri_Jun_14_16:34:21_PDT_2024
Cuda compilation tools, release 12.6, V12.6.20
Build cuda_12.6.r12.6/compiler.34431801_0
3.安装cuDNN
在官网下载: