CUDA(ComputeUnified Device Architecture)，是显卡厂商NVIDIA推出的运算平台。 CUDA是一种由NVIDIA推出的通用并行计算架构，该架构使GPU能够解决复杂的计算问题。当我们在进行深度学习任务时，我们有时需要使用GPU版本的框架，比如tensorflow-gpu或者mindspore-gpu版本就需要安装CUDA和CUDNN进行GPU加速支持

1.下载cuda10.1

CUDA Toolkit 10.1官方下载地址

在这里插入图片描述

2.下载cudnn7.6.4

cuDNN官方下载地址

注意要下载与cuda版本对应的cudnn工具

在这里插入图片描述

3.安装cuda10.1

sudo bash cuda_10.1.105_4188.39_linux.run

看到如下界面，在控制台输入accept进行下一步安装

在这里插入图片描述

然后再下述界面只需要勾选CUDA Toolkit 10.1即可，第一个为显卡驱动，如果当前驱动版本较低需要勾选安装

在这里插入图片描述

4.添加软链接

sudo cp ./include/* /usr/localcuda/include
sudo cp ./lib64/* /usr/local/cuda/lib64

5.配置环境变量

下图是已经安装好的cuda路径
在这里插入图片描述

vim ~/.bashrc

#cuda 配置cuda的安装路径，默认是在/usr/local
export CUDA_HOME=/usr/local/cuda-10.1
export LD_LIBRARY_PATH=${CUDA_HOME}/lib64
export PATH=${CUDA_HOME}/bin:${PATH}

source ~/.bashrc

6.验证是否安装成功

nvcc -V

在这里插入图片描述

如果出现该提示，说明安装成功

# 人工智能 # 深度学习 # 神经网络
Logo
昇腾开源生态专区

昇腾计算产业是基于昇腾系列（HUAWEI Ascend）处理器和基础软件构建的全栈 AI计算基础设施、行业应用及服务，https://devpress.csdn.net/organization/setting/general/146749包括昇腾系列处理器、系列硬件、CANN、AI计算框架、应用使能、开发工具链、管理运维工具、行业应用及服务等全产业链

更多推荐

cover

DeepSeek 崩了 13 小时，不是故障，是 V4 在换引擎

avatar 昇腾开源生态专区

体系结构论文（107）：AscendOptimizer: Episodic Agent for Ascend NPU Operator Optimization

摘要：本文提出AscendOptimizer系统，针对华为Ascend NPU的AscendC算子优化难题，通过双阶段交替优化方法实现性能提升。系统将算子拆分为host侧tiling program和device侧kernel program：Stage I采用进化搜索优化tiling策略，利用硬件反馈探索可行解空间；Stage II通过"优化回退"机制从优质kernel反向构

avatar 昇腾开源生态专区

AtomGit模型托管与实验管理全指南

AtomGit模型托管指南：AI开发者的版本控制利器 摘要：本文介绍AtomGit平台针对AI开发者推出的模型托管与实验管理功能，解决传统Git无法有效管理大模型文件、实验参数分散等问题。AtomGit通过Git LFS大文件存储、模型卡片元数据记录、代码与模型版本关联等创新功能，实现AI项目的一体化管理。文章详细演示了如何创建模型仓库、配置Git LFS管理大文件、编写结构化模型卡片，并特别介绍

avatar 昇腾开源生态专区