cuda 11.6 pytorch安装

在安装之前,需要先配置GPU环境(安装CUDA和CudaNN)

  • 命令行输入nvidia-smi,查看驱动信息
    nvidia-smi

安装相应的CUDA 和CUDANN

验证:输入nvcc --version 或者nvcc -V 进行检查

nvcc --version
nvcc -V

在anaconda里创建环境

conda env list  查看已有环境
conda remove -n  py39gpu  --all   删除已有环境

创建环境 

conda create -n py39gpu python=3.9

激活环境

conda activate py39gpu

设置conda下载的时长大一点,然后运行安装命令

conda config --set remote_read_timeout_secs 600.0conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=11.6 -c pytorch -c conda-forge


安装总结

1、需要确定conda、cudatoolkit、cudnn、python、PyTorch、torchvision的版本对应。

2、我们可以看到cuda的版本是11.6,所以上面我选择的10.2、11.3等低版本的,因为版本低一点是可以的,但不能选择高于11.6的版本。

关于Anaconda

Anaconda是一个开源的包、环境管理器,可以用于在同一个电脑上安装不同版本的软件包,并能够在不同的环境之间切换,Anaconda不是语言,它只是python的一个集成管理工具或系统,我们只需要安装Anaconda就可以了,甚至不用单独去安装python。
我们在使用过程中会出现conda,实际上这是一个可执行的命令,它的重要的功能就是包管理与环境管理,包管理也就是下载众多packages和科学计算工具等,环境管理就是用户可以安装不同版本的python来进行快速切换。安装anaconda让我们省去了大量下载模块包的时间,更加方便。

PyTorch  NVIDIA的CUDA技术和cuDNN库

PyTorch的GPU版本利用了NVIDIA的CUDA技术,使得深度学习计算能够高效地在GPU上运行。使用GPU来执行深度学习计算可以显著加速计算,从而减少训练和推理时间。

CUDA是NVIDIA推出的一种通用并行计算架构,可以使GPU执行通用计算任务,而不仅仅是图形处理。在PyTorch中,可以使用CUDA来利用NVIDIA GPU的并行计算能力加速模型训练和推理。

cuDNN是NVIDIA专门为深度学习模型设计的一个库,它提供了高效的卷积操作和其他计算操作,可以进一步加速深度学习任务。在PyTorch中使用cuDNN来优化深度学习模型的性能。

总的来说,PyTorch的GPU版本通过与NVIDIA的CUDA技术和cuDNN库的深度集成,为深度学习研究和应用提供了强大、灵活且高效的计算能力。

其他安装方法(我未能成功)

1、换清华源

pip --default-timeout=60000 install  torch==1.13.0+cu116 torchvision==0.14.0+cu116 torchaudio==0.13.0 --index-url https://download.pytorch.org/whl/cu116 -i https://pypi.tuna.tsinghua.edu.cn/simple

 清华源

https://pypi.tuna.tsinghua.edu.cn/simple

2、conda下载  清华源

conda install pytorch==1.13.0 torchvision==0.14.0 torchaudio==0.13.0 cudatoolkit=11.6 -c pytorch -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/

最后下成了cpu版本

3、离线下载方式一  (.whl)

确保已经安装好CUDA和CUDNN前提下,访问https://download.pytorch.org/whl/torch/,这里给出了所有版本的pytorch。但是页面的内容太多了不太容易找到我们需要的文件;如果确定用哪个版本的CUDA的话也可以访问下面这个网址:

https://download.pytorch.org/whl/cu+你的cuda版本号
例如cuda11.6,就访问https://download.pytorch.org/whl/cu116

依次将torch torchvision torchaudio三个.whl文件进行下载,下载好的文件如下:

下面正式进入whl安装,需要在虚拟环境中进入到刚才那三个.whl文件下载的路径 

如果当前的路径还在C盘下,而三个.whl文件下载的路径在D盘的话要先输入E盘的盘符E: 
其他下在非C盘的以此类推,否则直接输入cd +路径是无法成功进入目标路径的,如果下载在C盘则直接cd即可

比如我的.whl文件下载在E:\software则依次输入以下命令:

E:
cd E:\software

 先安装torch,输入pip install torch然后按Tab键自动补全:

pip install "torch-1.13.0+cu116-cp39-cp39-win_amd64.whl"

 再安装torchvision,输入pip install torchvision然后按Tab键自动补全:

最后安装torchaudio,输入pip install torchaudio然后按Tab键自动补全:

(----------没有成功-------)

4、离线下载方式二 (conda)

conda清华源找到对应的pytorch、torchvision、torchaudio的版本

清华大学开源软件镜像站链接如下:
链接: Index of /anaconda/cloud/pytorch/ | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror

 

conda install --offline torchaudio-0.13.0-py39_cu116.tar.bz2
conda install --offline torchvision-0.13.0-py39_cu116.tar.bz2
conda install --offline torchvision-0.14.0-py39_cu116.tar.bz2

出现了一个问题是

pip list里面有torch,torchaudio,torchvision。在项目文件里,import torch出错。

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.mzph.cn/bicheng/19715.shtml

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈email:809451989@qq.com,一经查实,立即删除!

相关文章

NLP课程笔记-基于transformers的自然语言处理入门

toc 项目地址 https://github.com/datawhalechina/learn-nlp-with-transformers/ 2017年,Attention Is All You Need论文(Google Brain)首次提出了Transformer模型结构并在机器翻译任务上取得了The State of the Art(SOTA, 最好)的效果。2…

ADB安装教程

1 adb简介 Android 调试桥 (adb) 是一种功能多样的命令行工具,可让您与设备进行通信。 adb命令可用于执行各种设备操作,例如安装和调试应用。 adb 提供对 Unix shell(可用来在设备上运行各种命令)的访问权限。它是一种客户端-服务…

HR人才测评,如何做中层管理人员的素质测评?

中层管理人员是企业中的重要力量,他们是连接高层管理和基层员工的桥梁,对企业的发展至关重要。因此,对中层管理人员的素质测评尤为重要。下面,我将介绍一些HR人才测评的方法和步骤,以帮助企业准确评估中层管理人员的素…

Jenkins从放弃到入门:部署、配置与应用

目录 Jenkins详解 一、Jenkins介绍 1、Jenkins 功能 2、Jenkins 概念 3、Jenkins 目的 4、Jenkins 特性 5、产品发布流程 二、Jenkins CI/CD 流程 三、部署Jenkins git 1、jenkins 安装 yum 安装 jenkins *jenkins 依赖 java 环境 #注意2.346之后的版本不再支持jdk…

RabbitMQ-发布/订阅模式

1、发布/订阅模式介绍 在普通的生产者、消费者模式,rabbitmq会将消息依次传递给每一个消费者,一个worker一个,平均分配,这就是Round-robin调度方式,为了实现更加复杂的调度,我们就需要使用发布/订阅的方式…

EXSI虚拟机新增磁盘并将空间扩充到已有分区

这里写自定义目录标题 1、在EXSI虚拟机中新增一块磁盘配置大小2、确认新磁盘3、格式化新分区4、添加新分区到LVM5、将新增分区添加到已有分区里 1、在EXSI虚拟机中新增一块磁盘配置大小 注意事项: (1)需确保虚拟机已关闭活处于维护模式,避免数据丢失 (2…

Spring Boot详解:深入了解与实践

文章目录 1. Spring Boot简介1.1 什么是Spring Boot?1.2 Spring Boot的历史背景1.3 Spring Boot的核心特点 2. Spring Boot的核心概念2.1 自动配置2.1.1 自动配置原理2.1.2 自定义配置 2.2 Spring Boot Starter2.3 Spring Boot CLI 3. Spring Boot的主要功能模块3.1…

网桥、路由器和网关有什么区别

在计算机网络领域,网桥、路由器和网关都是常见的网络设备,它们在网络通信中扮演着不同的角色。虽然它们都有连接不同网络的功能,但在实际应用中却具有各自独特的作用和特点。 1.网桥(Bridge) 定义:网桥是…

最佳 Mac 数据恢复:恢复 Mac 上已删除的文件

尝试过许多 Mac 数据恢复工具,但发现没有一款能达到宣传的效果?我们重点介绍最好的 Mac 数据恢复软件 没有 Mac 用户愿意担心数据丢失,但您永远不知道什么时候会发生这种情况。无论是意外删除 Mac 上的重要文件、不小心弄湿了 Mac、感染病毒…

【Epoch,Batch,Iteration】深度学习模型训练相关基础概念光速理解!

🔥模型训练相关基础概念! Epoch: 一次 epoch 代表整个训练数据集已经被完整地送入神经网络进行了一轮训练。通常,模型需要多次 epoch 才能充分学习数据集中的模式。Batch: 由于数据集可能过大,无法一次性全部加载到内存中进行训练…

解决VSCode右键没有Open In Default Browser问题

在VSCode进行Web小程序测试时,我们在新建的HTML文件中输入 !会自动生成页面代码骨架,写入内容后,我们想要右键在浏览器中预览。发现右键没有“Open In Default Browser”选项。原因是没有安装插件。 下面是解决方案:首先在VSCode找…

探索Lora:微调大型语言模型和扩散模型的低秩适配方法【原理解析,清晰简洁易懂!附代码】

探索Lora:微调大型语言模型和扩散模型的低秩适配方法 随着深度学习技术的快速发展,大型语言模型(LLMs)和扩散模型(Diffusion Models)在自然语言处理和计算机视觉领域取得了显著的成果。然而,这…

3d渲染的常用概念和技术,渲染100邀请码1a12

之前我们介绍了3D渲染的基本原理和流程,这次说下几个常用概念和技术。 3D渲染中涉及到很多专业的概念和技术,它们决定了渲染质量和效果,常用的有以下几个。1、光线追踪 光线追踪是一些专业渲染器(如V-Ray和Corona等)…

Django学习一:创建Django框架,介绍Django的项目结构和开发逻辑。创建应用,编写主包和应用中的helloworld

文章目录 前言一、Django环境配置1、python 环境2、Django环境3、mysql环境4、IDE:pycharm 二、第一次创建Django项目1、创建项目door_web_django_system2、运行启动 三、Django项目介绍1、介绍Django项目结构2、第一个helloword4、django的项目逻辑(和j…

React + Taro 项目 实际书写 感受

之前我总结了部分react 基础 根据官网的内容 以及Taro 框架的内容 今天我试着开始写了一下页面和开发 说一下我的感受 我之前写的是vue3 今天是第一次真正根据需求做页面开发 和逻辑功能 代码的书写 主体就是开发了这个页面 虽说这个页面 很简单 但是如果你要是第一次写 难说…

CATIA入门操作案例——压缩弹簧绘制,螺旋线的使用,法则曲线应用

目录 引出画压缩弹簧画等距部分画两端的压缩部分曲线缝合和扫掠封闭曲面得实体 总结异形弹簧新建几何体草图编辑,画一条样条线进行扫掠,圆心和半径画出曲面上的螺旋线再次选择扫掠,圆心和半径 其他自定义信号和槽1.自定义信号2.自定义槽3.建立…

Aigtek功率放大器的主要性能要求有哪些

功率放大器是电子系统中的重要组件,用于将低功率信号放大到高功率水平。功率放大器的性能直接影响到信号的放大质量和系统的整体性能。下面西安安泰将介绍功率放大器的主要性能要求。 增益:功率放大器应当具有足够的增益,即将输入信号的幅度放…

读取YUV数据到AVFrame并用多线程控制帧率

文件树: 1.xvideo_view.h class XVideoView { public:// 像素格式枚举enum Format { RGBA 0, ARGB, YUV420P };// 渲染类型枚举enum RenderType { SDL 0 };// 创建渲染对象的静态方法static XVideoView* Create(RenderType type SDL);// 绘制帧的方法bool DrawF…

影响生产RAG流水线5大瓶颈

检索增强生成(Retrieval Augmented Generation,RAG)已成为基于大型语言模型的生成式人工智能应用的关键组成部分。其主要目标是通过将通用语言模型与外部信息检索系统集成,增强通用语言模型的能力。这种混合方法旨在解决传统语言模…

无法删除dll文件

碰到xxxxxx.dll文件无法删除不要慌! 通过Tasklist /m dll文件名称 去查看它和哪个系统文件绑定运行,发现是explorer.exe。 我们如果直接通过del命令【当然需要在该dll文件所在的路径中】。发现拒绝访问 我们需要在任务管理器中,将资源管理器…