ubuntu 安装 nvidia anaconda cuda cudnn tensorflow pytorch

本文介绍了一个新系统的搭建流程,包括更换中科大源、安装NVIDIA显卡驱动、Anaconda35.2.0,以及在虚拟环境中安装TensorFlow和PyTorch等深度学习框架。
TensorFlow-v2.15

TensorFlow-v2.15

TensorFlow

TensorFlow 是由Google Brain 团队开发的开源机器学习框架,广泛应用于深度学习研究和生产环境。 它提供了一个灵活的平台,用于构建和训练各种机器学习模型

  1. 对于一个新系统
  2. 首先换源,建议使用中科大的源 参考链接
  3. 安装nvidia显卡驱动
  4. 安装anaconda3 5.2.0
  5. 安装tenssorflow和pytorch(注意建立虚拟环境)

您可能感兴趣的与本文相关的镜像

TensorFlow-v2.15

TensorFlow-v2.15

TensorFlow

TensorFlow 是由Google Brain 团队开发的开源机器学习框架,广泛应用于深度学习研究和生产环境。 它提供了一个灵活的平台,用于构建和训练各种机器学习模型

个人防护装备实例分割数据集 一、基础信息 • 数据集名称:个人防护装备实例分割数据集 • 图片数量: 训练集:4524张图片 • 训练集:4524张图片 • 分类类别: 手套(Gloves) 头盔(Helmet) 未戴手套(No-Gloves) 未戴头盔(No-Helmet) 未穿鞋(No-Shoes) 未穿背心(No-Vest) 鞋子(Shoes) 背心(Vest) • 手套(Gloves) • 头盔(Helmet) • 未戴手套(No-Gloves) • 未戴头盔(No-Helmet) • 未穿鞋(No-Shoes) • 未穿背心(No-Vest) • 鞋子(Shoes) • 背心(Vest) • 标注格式:YOLO格式,适用于实例分割任务,包含边界框或多边形坐标。 • 数据格式:图片数据,来源于监控或相关场景。 二、适用场景 • 工业安全监控系统开发:用于自动检测工人是否佩戴必要的个人防护装备,提升工作场所安全性,减少工伤风险。 • 智能安防应用:集成到监控系统中,实时分析视频流,识别PPE穿戴状态,辅助安全预警。 • 合规性自动化检查:在建筑、制造等行业,自动检查个人防护装备穿戴合规性,支持企业安全审计。 • 计算机视觉研究:支持实例分割、目标检测等算法在安全领域的创新研究,促进AI模型优化。 三、数据集优势 • 类别全面:覆盖8种常见个人防护装备及其缺失状态,提供丰富的检测场景,确保模型能处理各种实际情况。 • 标注精准:采用YOLO格式,每个实例都经过精细标注,边界框或多边形坐标准确,提升模型训练质量。 • 真实场景数据:数据来源于实际环境,增强模型在真实世界中的泛化能力和实用性。 • 兼容性强:YOLO格式便于与主流深度学习框架(如YOLO、PyTorch等)集成,支持快速部署和实验。
### 安装和配置 CUDAcuDNNPyTorch 的方法 #### 1. 确定硬件与驱动版本 在开始安装之前,需确认 GPU 驱动已正确安装并支持目标 CUDA 版本。可以通过以下命令检查当前显卡及其驱动版本: ```bash nvidia-smi ``` 如果未安装 NVIDIA 显卡驱动程序,则需要先完成其安装[^1]。 --- #### 2. 下载并安装 CUDA 工具包 访问官方 NVIDIA CUDA Toolkit 页面下载适合的版本。对于 Ubuntu 20.04,推荐通过 `.deb` 文件或 `runfile` 方式安装。以下是基于 APT 软件源的方式: 更新软件源列表: ```bash sudo apt update ``` 添加 NVIDIA 的 GPG 密钥及软件仓库地址: ```bash wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-keyring_1.0-1_all.deb sudo dpkg -i cuda-keyring_1.0-1_all.deb sudo apt-get update ``` 安装指定版本的 CUDA(例如 CUDA 11.8): ```bash sudo apt-get install -y cuda-11-8 ``` 设置环境变量以便于后续调用 CUDA 库文件: 编辑 `~/.bashrc` 或 `/etc/profile.d/` 中的脚本文件,追加如下内容: ```bash export PATH=/usr/local/cuda-11.8/bin:$PATH export LD_LIBRARY_PATH=/usr/local/cuda-11.8/lib64:${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}} ``` 使更改生效: ```bash source ~/.bashrc ``` 验证安装成功与否: ```bash nvcc --version ``` --- #### 3. 安装 cuDNN cuDNN 是深度学习框架的重要依赖库之一。可以从 NVIDIA 开发者网站获取对应 CUDA 版本的 cuDNN 压缩包。解压后复制到相应目录即可完成集成操作。 假设已经获得 cuDNN v8.x 对应 CUDA 11.8 的 tar.gz 归档文件: ```bash tar -xzvf cudnn-linux-x86_64-*.*.*-cuda*.tgz sudo cp cuda/include/* /usr/local/cuda/include/ sudo cp cuda/lib64/* /usr/local/cuda/lib64/ sudo ldconfig ``` 再次校验是否正常加载 cuDNN 动态链接库: ```bash ls /usr/local/cuda/lib64 | grep libcudnn.so ``` --- #### 4. 使用 Conda 创建虚拟环境并安装 PyTorch 创建一个新的 Anaconda 虚拟环境用于隔离开发需求: ```bash conda create -n torch_env python=3.9 conda activate torch_env ``` 按照引用中的具体指令来安装兼容版本的 PyTorch 及其他组件: ```bash conda install pytorch==2.0.0 torchvision==0.15.0 torchaudio==2.0.0 pytorch-cuda=11.8 -c pytorch -c nvidia ``` 测试安装效果以确保一切运行无误: ```python import torch print(torch.cuda.is_available()) # 输出 True 表明可用 print(torch.version.cuda) # 打印对应的 CUDA 版本号 ``` --- ### 注意事项 - 如果遇到任何错误提示,请仔细阅读报错信息并与实际安装路径对比调整。 - 不同版本间的匹配关系非常重要;建议始终查阅最新文档资料作为指导依据[^2]。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值