wsl安装深度学习基础环境

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

在官网下载：