首页 > 系统相关 >ubuntu安装cuda和cudnn,并测试tensorflow和pytorch库的与cuda的兼容性(2023年版)

ubuntu安装cuda和cudnn,并测试tensorflow和pytorch库的与cuda的兼容性(2023年版)

时间:2024-03-22 18:35:10浏览次数:38  
标签:torch cudnn cuda 2023 tensorflow local usr

lspci |grep -i nvidia查看nvidia设备,看到GPU
gcc --version检查是否安装上gcc软件包
根据官方文档指示,pip install torch==1.13.1+cu117 -f https://download.pytorch.org/whl/torch_stable.html,pip install torchaudio==0.13.1+cu117 -f https://download.pytorch.org/whl/torch_stable.html, pip install torchvision==0.14.1+cu117 -f https://download.pytorch.org/whl/torch_stable.html
于是下载CUDA11.7

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-keyring_1.0-1_all.deb
sudo dpkg -i cuda-keyring*.deb
sudo apt-get update
sudo apt-get -y install cuda-11-7

到cudnn的历史归档下载页面,点击Download cuDNN v8.8.0 (February 7th, 2023), for CUDA 11.x的Local Installer for Linux x86_64 (Tar),解压并进入cudnn目录

sudo cp cuda/include/cudnn*.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*

添加/usr/local/cuda/bin到PATH,/usr/local/cuda/lib64到LD_LIBRARY_PATH。nvcc -V获取CUDA编译器信息
安装最新的tensorflow库,注意匹配好tensorflow和cuda的版本(看到tensorflow和cuda对照表第一行还在用cuda11.2,我以为这张表好久没更新了,结果第一行就是最新版的tensorflow的依赖情况,google的动作有点儿慢啊,cuda11都到8了)。我安装的tensorflow-2.11.0编译时使用了cudnn8.2和cuda11.2,刚好和我之前安装的cuda和cudnn的大版本相同,能用。 验证tensorflow能否调用CUDA:import tensorflow as tf;tf.test.gpu_device_name(),打印出GTX1060的名字,参考自检测tensorflow是否可以使用GPU
验证pytorch能否调用CUDA:import torch;torch.__version__;torch.cuda.is_available(),打印True,参考自Win10下配置Pytorch-GPU(CUDA10.1)
本文的上一版:ubuntu安装cuda、cudnn和nvidia-docker

拓展阅读(安装TensorRT):Win11基于WSL2安装CUDA、cuDNN和TensorRT

本文创建于2022.5.25/21.02,修改于2023.3.2/14.47

标签:torch,cudnn,cuda,2023,tensorflow,local,usr
From: https://www.cnblogs.com/tellw/p/18090235

相关文章

  • 2023-5-11-elasticsearch使用
    索引操作、数据操作索引操作索引的创建、删除等创建索引PUT/shopping{"acknowledged":true,"shards_acknowledged":true,"index":"shopping"}获取索引详细信息GET/_cat/indices?vhealthstatusindexuuid......
  • 2023-5-3-skywalking基本使用
    服务探针配置、服务安装服务配置探针无侵入配置1.下载agent并解压skywalking-agent├──LICENSE├──NOTICE├──activations├──bootstrap-plugins├──config├──licenses├──logs├──optional-plugins├──optional-reporter-plugins├─......
  • 2023-8-10-canal使用
    工作原理、mysql开启二进制日志、启动服务端、启动客户端工作原理canal模拟MySQLslave的交互协议,伪装自己为MySQLslave,向MySQLmaster发送dump协议MySQLmaster收到dump请求,开始推送binarylog给slave(即canal)canal解析binarylog对象(原始为byte流......
  • 2023-6-20-springboot中使用es
    依赖、配置、定义索引对象、操作、其他依赖<!--Maven--><dependency><groupId>org.springframework.boot</groupId><artifactId>spring-boot-starter-data-elasticsearch</artifactId></dependency>配置spring:elasticsearch:......
  • 2023-12-22-flink-cdc使用
    应用场景、上手体验应用场景FlinkCDC(ChangeDataCapture)是一种用于捕获和处理数据源中的变化的流处理技术。它可以实时地将数据源中的增量更新捕获到流处理作业中,使得作业可以实时响应数据变化。以下是FlinkCDC的一些常见应用场景:数据仓库和实时分析:FlinkCDC可以......
  • 2023-12-5-logstash和filebeat使用
    应用场景、组件介绍、logstash启动、filebeat启动应用场景分布式场景中,不同服务器的服务日志集中收集管理,方便排查问题组件介绍logstash日志收集器,将接受到的日志存储到ES中fielbeat日志解析器,将日志解析后通过网络发送给日志收集器logstash启动下载https://www.elastic......
  • 2023年华数杯国际大学生数学建模竞赛B题社会稳定预警研究求解全过程文档及程序
    2023年华数杯国际大学生数学建模竞赛B题社会稳定预警研究原题再现  人类和所有动物一样,都有趋利避害的本能。人类成为造物之主的关键是,他们比其他动物更善于避免伤害。危机总是潜伏在未来。人类发展史是一部不断超越危机的历史(严耀军,2003)。“居安思危”,衡量和警示社会......
  • 退钱了!2023年个税扣缴退付申请倒计时~
       个人所得税手续费,大白话就是公司帮员工代扣代缴个税,税务机关取得的2023年个税总额2%的手续费退付。但是必须要企业去申请,3月31日前网上提交申请,税局就把钱返还给企业。过期不申请,视同放弃。    自2024年1月1日至3月31日,扣缴单位可通过自然人电子税务局(扣缴端)【......
  • 中国1KM分辨率月平均气温数据集1950-2023
       该数据为中国逐月平均温度数据,空间分辨率为0.0083333°(约1km),时间为1950年1月-2023年6月。该数据集是根据全国2472个气象观测点数据进行插值获取,验证结果可信。本数据集包含的地理空间范围是全国主要陆地(包含港澳台地区),不含南海岛礁等区域。数据坐标系统使用WGS84。自......
  • 中国30米分辨率城市建成区数据集 (1972-2023年)
      城市建成区的提取是近年来遥感应用领域研究的热点内容。传统的研究方法主要包括都市化指数(urbanindex,UI)、归一化建筑指数(normalizeddifferencebuilt-upindex,NDBI)、归一化植被指数(normalizeddifferencevegetationindex,NDVI)等方法。这些指数不仅可以单......