CUDA/CUDNN下载安装以及适配pytorch和tensorflow

CUDA以及CUDNN下载安装

在https://developer.nvidia.com/cuda-toolkit-archive可以找到各个版本的cuda,个人建议下载cuda11.0(也就是2年前的),太新或太旧都可能出现适配问题
在https://developer.nvidia.com/rdp/cudnn-archive下载符合cuda版本的cudnn压缩包,解压之后把里面的bin、include、lib文件夹等复制到cuda安装路径下(CUDA默认安装路径为C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0)

**Tips:**如果访问外网不稳定,可以去百度上直接搜cuda和cudnn,应该有很多云盘资源

适配Pytorch

之前编程的时候发现虽然CUDA安装好了,但是torch.cuda.is_available()还是返回False,这时候往往是版本不匹配.
解决办法就是在https://download.pytorch.org/whl/torch_stable.html找到对应版本的torch和torchvision,然后用pip install XXX.whl安装即可(如果网速过慢,可以试一下https://blog.csdn.net/Louise_Trender/article/details/103500509?spm=1001.2014.3001.5502中的方法,也可以到著名的第三方仓库找找]

适配Tensorflow

Windows系统

1.CPU版本:
在这里插入图片描述
2.GPU版本:
在这里插入图片描述

Linux系统

1.CPU版本:
在这里插入图片描述
2.GPU版本:
在这里插入图片描述

Conda

使用Conda安装,不需要安装单独安装cuda和cudnn:
conda install pytorch torchvision torchaudio cudatoolkit=11.0 -c pytorch

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值