debian系统用spack安装cuda和cudnn

spack——一个跨平台的软件包管理工具,docs链接如下:https://spack.readthedocs.io/en/latest/
服务器管理员提示use spack to load cuda,那咱就照做咯

spack install cuda

发现已经安装(主要是获取cuda root)
在这里插入图片描述

spack install cudnn

等待安装(装好了才来写的博客所以没截图),
若要安装特定版本

spack versions cudnn

然后选一个版本安装

spack install cudnn@版本号

后面cuda和cudnn的root我都用/cuda_root/,/cudnn_root/代替
然后把cuda环境变量加到自己用户环境变量配置文件中

vim ~/.bashrc

然后按i通过添加:

export PATH="/cuda_root//bin:$PATH"
export CUDA_HOME=$PATH:/cuda_root/
export LD_LIBRARY_PATH="/cuda_root//lib64:$PATH"

到最后一行
之后别忘了

source ~/.bashrc

ps: debian下vim粘贴不能用右键,咱也没有管理员权限没法改vim配置,只能用ctrl+insert组合键来复制,shift+insert组合键来粘贴
然后把/cudnn_root/include/下的cudnn.h复制到/cuda_root/include/

cp /cudnn_root/include/cudnn.h /cuda_root/include/

然后把/cudnn_root/lib64/目录下的lib开头文件都复制到/cuda_root/lib64

cp lib64/lib* /usr/local/cuda/lib64/ # 复制动态链接库

然后删了原来的动态文件

sudo rm -rf libcudnn.so libcudnn.so.7

在这里插入图片描述
就是这俩红色的文件
然后用libcudnn.so.7.6.5重新生成(后面版本号之后可能会不同)

ln -s libcudnn.so.7.0.5 libcudnn.so.7  # 生成软衔接
ln -s libcudnn.so.7 libcudnn.so  # 生成软链接

然后nvcc -V如下图
在这里插入图片描述
然后就可以了
另外,删除之前的pytorch-cpu之后再装gpu版本发现总是装成cpu版本
从conda新建一个环境就好了

conda create -n pytorch # 新建环境
conda activate pytorch # 切换环境
conda deactivate # 退出环境
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
Windows Subsystem for Linux (WSL2) 是 Windows 10 中的一项功能,它提供了一个在 Windows 上运行完整版 Linux 分区的环境。如果你想在 WSL2 中安装 CUDAcuDNN,这是一个通常用于深度学习的 GPU 加速库,你可以按照以下步骤操作: 1. **安装 NVIDIA GPU 驱动**: - 首先,确保你的 Windows 系统有支持 CUDA 的 NVIDIA GPU,并访问 NVIDIA 的官网下载适用于你的 GPU系统的最新驱动程序。 2. **启用开发者模式**: - 在设备管理器中,找到 NVIDIA GPU,右键点击选择“属性”>“高级”>“启用设备虚拟化”。 3. **安装 Ubuntu 或其他发行版**: - 从 Microsoft Store 或者通过命令行(`wsl --install -d <distro-name>`)安装 Ubuntu 或者其他你喜欢的 Linux 发行版,如 Debian 或者 CentOS。 4. **更新和安装依赖**: - 登录 WSL,运行 `sudo apt update && sudo apt upgrade` 更新系统。 - 安装 `cuda-toolkit` 和 `cudnn`,例如: ``` sudo apt install nvidia-cuda-toolkit sudo apt-get install libnvidia-cudnn-dev ``` 5. **配置环境变量**: - 需要设置 `LD_LIBRARY_PATH` 和 `CUDA_HOME` 环境变量指向正确的库目录。 6. **验证安装**: - 测试安装是否成功,可以尝试运行 `nvcc --version` 来查看 CUDA 版本,以及 `cudnn` 相关的测试工具。 **相关问题--:** 1. WSL2 支持哪些 Linux 发行版? 2. 如何在 WSL2 中查看已安装CUDA 版本? 3. 如何在 Ubuntu 中管理和更新软件包?
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值