首页 > 其他分享 >如何在anaconda环境中安装cuda

如何在anaconda环境中安装cuda

时间:2024-01-17 13:25:13浏览次数:36  
标签:头文件 anaconda cudnn cuda NVIDIA 安装

在前面的文章(几年前的文章)中我们介绍了在anaconda中安装cuda、cudnn后,有介绍了如何在anaconda中安装nvcc、nccl等NVIDIA的各种编译器和库,本文介绍如何在anaconda中安装NVIDIA的各种编译器和library库的头文件。

由于使用anaconda安装pytorch、TensorFlow、jax、mindspore比较方便,不需要手动的配置cuda和cudnn,但是有时候我们需要给各种的深度学习框架做扩展或者是源码编译,在之前的文章中给出了在anaconda环境下如何编译不同架构(x86/arm64)下的pytorch、TensorFlow、mindspore。

不论是给深度学习框架做源码编译还是做扩展library编译都需要用到(可能用到):cuda、cudnn、nvcc、nccl、头文件cuda.h、头文件cuda_runtime.h,为了能够在anaconda环境下进行这些操作我们需要在anaconda环境下手动安装这些支持,因为在anaconda的原生深度学习框架的官方安装命令中往往只有cuda和cudnn的安装。

关于nvcc、nccl的安装之前已经给出过安装方法,本文针对cuda的头文件的anaconda环境下安装。

(要知道,在anaconda下安装NVIDIA的各种依赖是anaconda公司已经预先匹配好版本号的,当然也可以手动去NVIDIA官方网站上下载对应版本并配置,只不过anaconda环境下更加简便)



为jax和pytorch编译扩展代码,报错:

image


提示缺少头文件:cuda.h


参考NVIDIA公司的GitHub讨论帖子:

地址:
https://github.com/NVIDIA/apex/issues/1591

image


得到具体的anaconda环境下的解决方法:

conda install nvidia::cuda-cudart-dev




标签:头文件,anaconda,cudnn,cuda,NVIDIA,安装
From: https://www.cnblogs.com/devilmaycry812839668/p/17969795

相关文章

  • Flume安装配置
    简介ApacheFlume是一个分布式、可靠、高可用的服务,用于有效地收集、聚合和移动大规模数据。Flume主要用于大数据环境下的日志收集和数据传输,能够帮助用户实现数据流的可靠传输和集中处理。flume安装配置解压安装包tar-zxvfapache-flume-1.9.0-bin.tar.gz-C/opt/module......
  • docker环境下安装RabbitMQ
    环境系统为debian12将安装docker将安装rabbitmq3.8.5安装步骤1Docker安装1.1Debian下安装非常简单:aptinstalldocker.io1.2查看docker的版本dockerversion1.3修改源vi/etc/docker/daemon.json{"registry-mirrors":["https://registry.docker-cn.com"]......
  • 一、selenium环境安装+元素定位
    1、html元素构成1)元素的构成<标签名属性名=属性值属性名=属性值>文本内容</标签名>2)通用属性idnameclassstyle 3)如何去查找元素:根据元素本身的特征去找-筛选 2、元素定位selenium49大元素定位策略--传统8大定位策略+selenium4新增第......
  • Linux系统安装Jmeter
     今天我们学习下如何在Linux下安装Jmeter,因为Jmeter是Java开发的,需要依赖JDK环境,因此我们需提前安装好JDK。本人环境:CentOS7.364位JDK1.8(可参考文章:Linux下安装JDK1.8)下载Jmeter安装包Jmeter是开源的工具,我们直接到官网下载即可,本次安装的版本是 5.6.3。最新版本......
  • VMware虚拟机安装openKylin(开放)麒麟系统
    1.麒麟系统简单介绍常见的有银河麒麟,中标麒麟,开放麒麟,优麒麟麒麟软件官网:https://www.kylinos.cn/开放麒麟官网:https://www.openkylin.top/优麒麟官网:https://www.ubuntukylin.com/开放麒麟和银河麒麟桌面操作系统的关系类似于深度deepin和统信UOS。前者为社区版本,后者是......
  • nmon安装
    我们在做性能测试的时候,是需要监控后端指标的,lr和jmeter提供了插件可以方便我们做这些,但是其中是有些不足的。比如获取的信息不全面或者存在兼容性问题nomn是ibm的开发的检测工具,小巧占用资源少,界面交互友好,且提供的监控指标很全面。安装:在linux系统上的安装,我是ubun......
  • 麒麟linux安装net8并配置netcore项目网站运行
     1.安装net8执行下面命令。从微软官方下载net8:https://dotnet.microsoft.com/zh-cn/download/dotnet/8.0https://dotnet.microsoft.com/zh-cn/download/dotnet/thank-you/sdk-8.0.101-linux-x64-binaries  然后进入linux,按照官方说明执行: mkdir-p$HOME/dotnet......
  • 基于centos环境的docker安装redis以及设置密码
    本篇的博客是因为之前的博客,写的存在问题,设置的密码无法使用,就是即使设置了密码,依旧提示认证失败,所以才有了本篇的博客文章 dockerpullredis 查看下载的镜像dockerimages新建redis配置文件mkdir-p/home/redis/conftouch/home/redis/conf/redis......
  • Androidstudio copilot 安装
    如何安装AndroidStudioCopilot1.安装AndroidStudio首先,你需要安装AndroidStudio,这是一个用于开发Android应用程序的集成开发环境(IDE)。你可以从官方网站(Studio,并按照安装向导的指示进行安装。2.配置AndroidStudio安装完成后,你需要对AndroidStudio进行一些基本配置。打......
  • IntelliJ IDEA 2024年最新下载、安装使用教程、亲测可用
    本文讲的是2023.3最新版本IntelliJIDEA破解、IntelliJIDEA激活码、IntelliJIDEA安装、IntelliJIDEA永久激活码的最新永久激活教程,本文有mac和windows系统的idea安装教程。1.下载安装IntelliJIDEA首先先到官网下载最新版的IntelliJIDEA,下载后傻瓜式安装就好了官网下载......