ubuntu cuda+cudnn

本文详细介绍了在Ubuntu系统中如何安装CUDA和CUDNN,以及后续的PyTorch环境配置。包括驱动检查、CUDA与CUDNN的下载与匹配、显卡信息查看、PyCharm与Anaconda的安装、虚拟环境创建和PyTorch的版本选择安装步骤。
摘要由CSDN通过智能技术生成

1.驱动安装

首先先确定是否安装驱动,驱动下载地址,一般情况下,驱动已经有了,这一步可以直接跳过。

2.cuda安装

注意在下载cuda的时候,注意版本,因为pytorch一般不支持最新版本,所以尽量不要下载cuda最新版本,cuda下载地址,将cuda下载之后,需要安装cuda,安装时需要命令行进行的,使用sudo sh 文件名,如图:cuda安装图片
接下来基本上选择yes或者接受,除过有一个选项时需要安装驱动的,选拒绝就可以了。

3.cudnn安装

在下载cudnn的时候,需要下载与cuda对应的版本,cudnn下载链接,下载之后,将cudnn中的文件复制到cuda对应的文件夹中(类似于windows中cudnn的方式),只不过ubuntu是需要命令行进行复制的。

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值