(如何安装pytorch详细教程!!!)全面讲解安装Anaconda+CUDA+PyTorch

2023-11-01

之前安装了CUDA,与Pytorch,但是当把自己的数据放到GPU上训练的时候(代码部分已经写清楚放到CUDA上),发现GPU的占用率很低,CPU的占用率高达90%,后来发现是显卡的驱动与自己安装的CUDA并不匹配,pytorch与cuda的版本也不匹配。在此记录下自己查了很多资料得到的理解。

1.安装Anaconda

进入官网Anaconda选择对应的版本。Anaconda:简单来说是用来管理python工具的,里面包含python编译器,可以用来安装各种插件。如果把Python类比Linux,那么Anaconda就是centos、ubuntu之类的。

选择对应的版本

安装AnaConda没有什么特殊的要求。但是需要记住以下几点:1.Anacaonda安装占用许多空间,尽量不要安装在C盘,路径不要有中文!! 2.勾选下面的选项,方便实现在控制台运行conda命令。

安装Anaconda

检查安装是否成功。出现一下内容表示安装成功。

Anaconda安装成功
2.安装CUDA,(非常重要,显卡比较新的话,cuda也会默认比较新,但是最新版本的pytorch是不支持最新版本的cuda的)(不支持AMD显卡,显卡是AMD请跳过此步)
可以使用nvidia-smi查看自己目前的cuda版本,如果cuda版本与pytorch支持的版本不一致的话,需要重新安装cuda,具体请看下面说明

①查看显卡型号,支持的CUDA最高版本(非常重要) 桌面右键→NVIDIA控制面板→帮助→系统信息→组件
由于,显卡年代久远,无法更新到最新的驱动程序,我的显卡最高支持11.1版本的CUD
CUDA与英伟达驱动对应支持版本,我的显卡驱动版本为457.49 > 456.81。选择此版本
在这里插入图片描述
对照关系查看[https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html]
在这里插入图片描述

②.下载CUDA Toolkit
这里有一个坑,在下载了高版本的CUDA后,发现Pytorch不支持最新版本的CUDA,去官网最新的Pytorch支持的CUDA版本,截止到目前只支持11.1版本的CUDA,所以只能下载11.1以及一下版本的CUDA!!!!

在这里插入图片描述

选择 “自定义©”,只安装 CUDA 开发套装部分,其他的不安装。如下图所示:

自定义CUDA
在这里插入图片描述

一直点下一步即可安装完成。大约需要3分钟。 默认安装目录为 C:\Program Files\NVIDIA GPU Computing
Toolkit\CUDA\v11.0,示例程序安装目录为 C:\ProgramData\NVIDIA Corporation\CUDA
Samples\v11.0。

安装完成后,在命令行窗口(cmd)中,输入 nvcc -V 命令进行测试。

安装CUDA成功

③.安装与下载并安装与 CUDA 版本兼容的 cuDNN(非必须),用于加速训练。

CUDA看作是一个工作台,上面配有很多工具,如锤子、螺丝刀等。cuDNN是基于CUDA的深度学习GPU加速库,有了它才能在GPU上完成深度学习的计算。它就相当于工作的工具,比如它就是个扳手。但是CUDA这个工作台买来的时候,并没有送扳手。想要在CUDA上运行深度神经网络,就要安装cuDNN,就像你想要拧个螺帽就要把扳手买回来。这样才能使GPU进行深度神经网络的工作,工作速度相较CPU快很多。

具体可参考CUDA与cuDNN的关系
下载对应版本 cuDNN 需要注册。
在这里插入图片描述
首先解压 cudnn-11.1-windows-x64-v8.0.5.39.zip,解压后的文件目录如下所示
在这里插入图片描述
然后,将所有内容复制到 cuda 11.0 安装目录,即可完成 cuDNN 的安装,如下图所示。
在这里插入图片描述
在这里插入图片描述

3.conda 安装 pytorch

由于正常安装pytorch太慢需要添加下载源,以管理员身份运行CMD
查看、删除、添加源更多详情请看传送门

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --add channels https://mirrors.sjtug.sjtu.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.sjtug.sjtu.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.sjtug.sjtu.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes

复制Pytorch官网的conda安装命令,删去 -c pytorch -conda -forge即可自动从其他源下载安装包:

在这里插入图片描述在这里插入图片描述在这里插入图片描述
4.查看是否安装成功
python

import torch
print(torch.__version__) #查看pytorch版本
print(torch.cuda_version) #查看cuda版本
print(torch.cuda.is_available()) #查看cuda是否可用

安装成功
5.本人新手,看了很多安装教程,很多都不全面,让我在安装过程中出现了很多问题,于是我查阅了很多资料,自己写了一个完整版的记录自己的学习。包括自己对安装过程中的理解。可能会有错误的地方,欢迎补充,补充。

本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)

(如何安装pytorch详细教程!!!)全面讲解安装Anaconda+CUDA+PyTorch 的相关文章

随机推荐

  • SPI技术-JDK实现

    SPI是什么 SPI全称Service Provider Interface 是Java提供的一套用来被第三方实现或者扩展的API 它可以用来启用框架扩展和替换组件 Java SPI 实际上是 基于接口的编程 策略模式 配置文件 组合实现的
  • OpenSSL创建生成CA证书、服务器、客户端证书及密钥

    使用OpenSSL创建生成CA证书 服务器 客户端证书及密钥 目录 使用OpenSSL创建生成CA证书 服务器 客户端证书及密钥 一 生成CA证书 二 生成服务器证书 三 生成客户端证书 四 测试 本片博客对对应资源下载 免积分 https
  • js轮播图的制作详细过程

    JS轮播图 1 在body里面写简单的HTML代码 首先在顶部head标签里引入外部CSS和JS 注意要引入JQuery库 并且放在所写的js上面 不然不会有效果 然后在一个div中包裹两个url 第一个用于显示图片 第二个用于定义页码 代
  • 数据结构——>栈

    栈 栈的介绍 栈的应用场景 栈的代码实现 实现栈的思路分析 入栈 出栈 遍历栈 栈的介绍 1 栈是一个先入后出的有序列表 想象成弹夹 2 变化的一端为栈顶 固定的一端为栈底 3 入栈演示图 4 出栈演示图 栈的应用场景 1 递归 2 四则运
  • Linux系统ioctl使用示例 ioctl实验

    本文是我在andoid实验的ioctl的功能 如双向传递参数 贴出来希望对学习ioctl的人很有帮助 linux的ioctl功能是很强大的 android显示模块还有camera模块都离不开ioctl让上层和内核交互 这个实验对初学者很有帮
  • malloc 背后的虚拟内存 和 malloc实现原理

    面试的时候经常会被问到 malloc 的实现 从操作系统层面来说 malloc 确实是考察面试者对操作系统底层的存储管理理解的一个很好的方式 涉及到虚拟内存 分页 分段等 下面逐个细说 1 虚拟内存 首先需要知道的是程序运行起来的话需要被加
  • 多态的作用

    对象的上转型对象体现了多态 多态的作用 将不同的子类对象统一看做父类 屏蔽了不同子类之间的差异 有子类到父类 有特殊到一般 是程序变得通用 public class TestPloy public sataic void main Stri
  • flink连接kafka报:org.apache.kafka.common.errors.TimeoutException

    测试flink1 12 7 连接kafka package org test flink import org apache flink api common serialization SimpleStringSchema import
  • 用chatgpt超级写手批量制作抖音带货文案并直接SEO霸屏

    带货文案怎么写 最好的解决方案就是学习同行 简版流程 通过抖音热点宝 手机端 的种草视频 或者抖音筛选出最多点赞 半年内的带货视频 提取出带货文案 然后用chatgpt直接改写一下带货文案 也可以根据抖音商城的爆款商品 提炼出卖点 然后利用
  • HTML5-表单

    HTML5 表单 一 Form 1 action 属性 action 属性用于指定表单提交时向何处发送表单数据 即需要发送的服务器地址
  • linux bash如何判断脚本命令参数是否存在

    linux bash处理较简单的参数 如 test sh 10 而很少使用 test n 10这种带选项的方式 典型用法为 bin bash if x 1 x then 有参数 else then 没有参数 fi 为什么要使用 x 1 x
  • MII接口及应用

    MII是英文Medium Independent Interface的缩写 翻译成中文是 介质独立接口 该接口一般应用于以太网硬件平台的MAC层和PHY层之间 MII接口的类型有很多 常用的有MII RMII SMII SSMII SSSM
  • sink 写入本地文件(八)

    sink 写入本地文件 2 4 Sink 2 4 1 将数据写入文件 File Sink 2 4 Sink Flink 没有类似于spark中 foreach 方法 让用户进行迭代的操作 虽有对外的输出操作都要利用Sink完成 最后通过类似
  • 软件工程 第一次随堂练习

    以下答案是经过人工智能生成 个人理解得出的答案 若有不同见解 请在评论区留言或私信 说明下列需求分别属于下面的哪种类型 为什么 A 业务需求 B 用户需求 C 系统级 功能 需求 D 性能需求 E 质量需求 F 约束 G 对外接口 H 数据
  • linux的aix下的makefile参考

    一 详解 1 简单模板makefile CC OBJS CFLAGS TARGET all TARGET TARGET OBJS CC CFLAGS o OBJS SUFFIXES o h SUFFIXES cpp o cpp o CC c
  • APP首次进入,弹出服务协议和隐私政策

    app上应用市场里需要几个条件 需要第一次下载并进入首页前展示用户协议和隐私政策弹窗提醒 以及在应用内版块有常驻入口 就是应用里有可以看到这个协议的版块 登录或者注册页面也需要有同意服务协议和隐私政策的提醒 在上个项目上线时候遇到andro
  • 服务器管理口安装系统,管理口安装服务器操作系统

    管理口安装服务器操作系统 内容精选 换一换 如果Windows操作系统云服务器未安装密码重置插件 可以参见本节内容重新设置密码 本节操作介绍的方法仅适用于修改Windows本地账户密码 不能修改域账户密码 Linux操作系统请参见重置Lin
  • Golang 从零开始实现多人聊天室(五)开发前奏-目录结构优化

    系列文章目录 跟着 猫猫学Golang 快人一步 系列初开 跟着我走进Go 语言的世界里 系列目录 Golang 从零开始实现多人聊天室 一 服务端监听 Golang 从零开始实现多人聊天室 二 客户端访问 Golang 从零开始实现多人聊
  • 图像阈值cv2.threshold

    图像阈值 图像阈值函数 图像阈值函数 ret dst cv2 threshold src thresh maxval type src 输入图 只能输入单通道图像 通常来说为灰度图 dst 输出图 thresh 阈值 maxval 当像素值
  • (如何安装pytorch详细教程!!!)全面讲解安装Anaconda+CUDA+PyTorch

    之前安装了CUDA 与Pytorch 但是当把自己的数据放到GPU上训练的时候 代码部分已经写清楚放到CUDA上 发现GPU的占用率很低 CPU的占用率高达90 后来发现是显卡的驱动与自己安装的CUDA并不匹配 pytorch与cuda的版