CUDA and cuDNN 配置

CUDA

CUDA是NVIDIA推出的用于自家GPU的并行计算框架,也就是说CUDA只能在NVIDIA的GPU上运行,而且只有当要解决的计算问题是可以大量并行计算的时候才能发挥CUDA的作用。

cuDNN

cuDNN(CUDA Deep Neural Network library):是NVIDIA打造的针对深度神经网络的加速库,是一个用于深层神经网络的GPU加速库。如果你要用GPU训练模型,cuDNN不是必须的,但是一般会采用这个加速库。

CUDA Driver Version和CUDA Runtime Version的区别

        GPU Driver, CUDA 和 PyTorch的版本关系

CUDA Toolkit 安装

  • 安装之前首先要确认你需要安装的CUDA和cuDNN的版本
    • 首先确定CUDA的版本
    • 然后选择一个与之适配cuDNN版本
  • root权限同时安装多个版本的cuda,并且进行配置
  • 非root权限从已安装的CUDA版本中选择合适的版本,在修改环境变量进行配置
  1. 切换CUDA的两种方式

    1. 修改 ~/.bashrc 中CUDA的环境变量(推荐)
    2. 修改cuda库的软连接(root权限)
  2. 官网下载目标版本的CUDA

    Installer Type 选择 runfile(local),比较方便

    下载地址:http
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值