win10中CUDA cundnn pytorch环境搭建记录

news2024/9/26 5:12:09

关于在win10中安装cuda cudnn及pytorch全家桶(torch torchvision torchaudio)的详细安装步骤,可以参考这个帖子,说的非常详细!

win10下pytorch-gpu安装以及CUDA详细安装过程

仅在此记录一下我的两台电脑安装的环境

目录

  • 一、笔记本环境配置
    • 1.根据显卡型号查看适配的CUDA版本
    • 2.下载CUDA并配置环境变量
    • 3.根据CUDA下载并解压cudnn
    • 4.根据CUDA安装pytorch
  • 二、台式机环境配置
    • 1.根据显卡型号查看适配的CUDA版本
    • 2.下载CUDA并配置环境变量
    • 3.根据CUDA下载并解压cudnn
    • 4.根据CUDA安装pytorch
  • 卸载

一、笔记本环境配置

1.根据显卡型号查看适配的CUDA版本

显卡是2080 Super
在这里插入图片描述
从显卡算力查询网站,我们可以查询到自己的显卡支不支持CUDA(存在即支持)和显卡的算力。虽然这上面没有super但是有2080其实也就够了。

算力查询网站

在这里插入图片描述

在这里插入图片描述
在cmd中输入nvidia-smi,查询适配自己显卡的CUDA版本,笔记本对应的是CUDA11.6
在这里插入图片描述

2.下载CUDA并配置环境变量

我选择下载的CUDA11.6是这个:cuda_11.6.2_511.65_windows.exe(下载11.6的就行,最后的数字应该是补丁,感觉大小版本一致的前提下补丁越多越好)。

CUDA下载网址

在这里插入图片描述
在这里插入图片描述
下载完成exe后安装即可,安装在默认位置(C盘),安装选项为自定义安装,不选Visual Studio Intergration(不过我下载的这个exe并没有这个)

在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
安装完成后,找到环境变量-系统变量-Path,点击编辑:
在这里插入图片描述

新建以下四个环境变量:
在这里插入图片描述

记得一定要点击确定,否则环境变量不生效。

修改完后,cmd测试一下,显示以下输出就说明CUDA安装成功了

nvcc -V

在这里插入图片描述
要是提示nvcc命令不是内部或外部命令,也不是可运行的程序。
首先排查下是不是CUDA安装失败了:
cmd打开安装位置:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.7\bin
在这里插入图片描述

在这里输入nvcc -V,如果在这输入后还是找不到nvcc,说明CUDA没安装成功;如果找得到nvcc,并且输出正常信息,说明环境变量搞错了。查看是不是之前安装过其他版本的CUDA,环境变量没改回来。

3.根据CUDA下载并解压cudnn

cudnn我选择的是这个:cudnn-windows-x86_64-8.9.2.26_cuda11-archive.zip

cudnn下载网址

在这里插入图片描述

在这里插入图片描述
下载完成后,解压,全部复制粘贴到CUDA目录下:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6

4.根据CUDA安装pytorch

pytorch安装的是以前的版本,适配于CUDA11.6:

pytorch下载网址

在这里插入图片描述
使用官网上提示的pip自动安装指令,在anaconda中安装最新的适配于CUDA11.6的torch全家桶

pip install torch==1.13.1+cu116 torchvision==0.14.1+cu116 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu116

在这里插入图片描述
然而用这种方法去安装,下载速度非常慢。所以可以根据conda终端显示的下载地址,直接打开下载链接

https://download.pytorch.org/whl/cu116/torch-1.13.1%2Bcu116-cp38-cp38-win_amd64.whl

这样就会用浏览器下载torch-1.13.1+cu116-cp38-cp38-win_amd64.whl

下载完成后,随便放一个位置(笔记本我放在了D:\Anaconda3\envs\DRL\Scripts)

在conda中激活DRL环境,使用pip离线安装:

pip install torch-1.13.1+cu116-cp38-cp38-win_amd64.whl

这样最大的torch就安装完了,继续执行pip自动安装指令

pip install torch==1.13.1+cu116 torchvision==0.14.1+cu116 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu116

将自动安装适配的torchvision和torchaudio,出现报错大概率是网络问题,重试几遍即可。

安装完毕,查看pytorch是否可用:

import torch
print(torch.__version__)
print(torch.cuda.is_available())

在这里插入图片描述
出现上图提示信息,说明pytorch安装完毕,并且CUDA可用。

二、台式机环境配置

1.根据显卡型号查看适配的CUDA版本

显卡是3090

在这里插入图片描述
众所周知,3090是必有CUDA的

算力查询网站

在这里插入图片描述

在这里插入图片描述
在cmd中输入nvidia-smi,查询适配自己显卡的CUDA版本,台式机对应的是CUDA11.7
在这里插入图片描述

2.下载CUDA并配置环境变量

我选择下载的CUDA11.7是这个:cuda_11.7.1_516.94_windows.exe(下载11.7的就行,最后的数字应该是补丁,感觉大小版本一致的前提下补丁越多越好)。

CUDA下载网址

在这里插入图片描述

在这里插入图片描述

安装过程和笔记本是一样的
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述

安装完成后,找到环境变量-系统变量-Path,点击编辑:
在这里插入图片描述
新建以下四个环境变量:
在这里插入图片描述

记得一定要点击确定,否则环境变量不生效。

修改完后,cmd测试一下,显示以下输出就说明CUDA安装成功了

nvcc -V

在这里插入图片描述

3.根据CUDA下载并解压cudnn

cudnn我选择的是这个:cudnn-windows-x86_64-8.9.2.26_cuda11-archive.zip

cudnn下载网址

在这里插入图片描述

在这里插入图片描述
下载完成后,解压,全部复制粘贴到CUDA目录下:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.7

4.根据CUDA安装pytorch

pytorch依旧采用pip安装,复制自动安装指令:

pytorch下载网址

在这里插入图片描述

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu117

和笔记本一样,这种方法安装下载速度非常慢。依旧可以根据conda显示安装pytorch对应whl的下载地址,直接打开下载链接

https://download.pytorch.org/whl/cu117/torch-2.0.1%2Bcu117-cp38-cp38-win_amd64.whl

这样就会直接用浏览器下载torch-2.0.1+cu117-cp38-cp38-win_amd64.whl
在这里插入图片描述

下载完成后,随便放一个位置(台式机放在了D:\CUDA
在conda中激活DRL环境,使用pip离线安装:

pip install torch-2.0.1+cu117-cp38-cp38-win_amd64.whl

但是我也不知道台式的网络是怎么回事,在安装torch-2.0.1+cu117-cp38-cp38-win_amd64.whl的依赖包时一直会因为网络原因安装失败。所以只能用清华源镜像安装依赖包了。也就是把上述的安装方式换成:

pip install torch-2.0.1+cu117-cp38-cp38-win_amd64.whl -i https://pypi.tuna.tsinghua.edu.cn/simple

这样最大的torch就安装完了,继续执行自动安装指令

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu117

将自动安装适配的torchvision和torchaudio,出现报错大概率是网络问题,重试几遍即可。(重试还是不行,建议直接换清华源)

安装完毕,查看pytorch是否可用:

import torch
print(torch.__version__)
print(torch.cuda.is_available())

在这里插入图片描述

说明torch安装完毕,并且CUDA可用。

卸载

非常简单,用geek卸载掉这几个文件
在这里插入图片描述

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

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

相关文章

Leetcode-每日一题【剑指 Offer II 009. 乘积小于 K 的子数组】

题目 给定一个正整数数组 nums和整数 k ,请找出该数组内乘积小于 k 的连续的子数组的个数。 示例 1: 输入: nums [10,5,2,6], k 100输出: 8解释: 8 个乘积小于 100 的子数组分别为: [10], [5], [2], [6], [10,5], [5,2], [2,6], [5,2,6]。 需要注意的是 [10,5,2]…

代码-【5 二叉树非递归后序遍历,找指定结点的父节点】

二叉树T按二叉链表存储,求指定结点q的父节点:

通过一次线上问题,讲下Ribbon重试机制

前言 前段时间,产品经理在线上验证产品功能的时候,发现某个功能不符合需求预期,后来测试验证发现是服务端的一个接口大概率偶现超时,前端做了兜底处理,所以对线上用户么有太大影响。 问题排查过程 由于服务端的接口…

【暑期每日一练】 day8

目录 选择题 (1) 解析: (2) 解析: (3) 解析: (4) 解析: (5) 解析: 编程题 题一 描述…

简单聊聊创新与创造力

文章目录 前言一、大脑运行的两种方式1、聚焦模式2、发散模式3、影响想法的因素a、背景知识b、兴趣c、天赋 4、思维固化 二、想法的不可靠1、对想法进行验证2、颠覆性创新,挤牙膏式创新3、为什么模仿这么多 三、更多更多的idea1、个人的方面a、积累不同的背景知识b、…

怎么在线修改图片?分享一个图片修改工具

无论是在个人或商业领域,我们都需要使用高质量的图片来传达信息或提高品牌形象。大尺寸的图片也会占据大量的存储空间和带宽,影响网站的加载速度和用户体验。因此,我们需要一种高效的工具来解决这个问题。今天向大家介绍一款非常实用的图片处…

(学习笔记-内存管理)内存分段、分页、管理与布局

内存分段 程序是由若干个逻辑分段组成的,比如可由代码分段、数据分段、栈段、堆段组成。不同的段是有不同的属性的,所以就用分段的形式把这些分段分离出来。 分段机制下,虚拟地址和物理地址是如何映射的? 分段机制下的虚拟地址由…

JetBrains 为测试自动化打造的强大 IDE-Aqua

QA 和测试工程对现代软件开发必不可少。 在 JetBrains,我们相信使用正确的工具对每项工作都很重要。 对我们来说,为自动化测试开发创建单独的工具是自然而然的事,因为这使我们能够满足多角色软件开发团队的需求。 我们很高兴能够推出 JetBra…

【Vscode】远程内存占用大

查看远程服务器上的扩展 依次删除,重新连接后观察内存占用 此扩展占用较高,约2G(前后端项目,依赖较多导致)

JS前端读取本地上传的File文件对象内容(包括Base64、text、JSON、Blob、ArrayBuffer等类型文件)

读取base64图片File file2Base64Image(file, cb) {const reader new FileReader();reader.readAsDataURL(file);reader.onload function (e) {cb && cb(e.target.result);//即为base64结果}; }, 读取text、JSON文件File readText(file, { onloadend } {}) {const re…

一个写了3年半flutter的小伙,突然写了2个月uniapp的感悟!

前言 因为某些原因,在过去的三年半时间,我除了flutter之外,很少接触其他的框架,期间除了学习了Android(主要是Kotlin、jetpack)、GoLang Gin之外基本上很少接触其他的框架。而在最近的两个月,突…

如何基于 Apache Doris 构建新一代日志分析平台

作者:肖康,SelectDB 技术 副总裁 、Apache Doris Committer 日志数据是企业大数据体系中重要的组成部分之一,这些数据记录了网络设备、操作系统以及应用程序的详细历史行为,蕴含了丰富的信息价值,在可观测性、网络安全…

项目篇:Echo论坛系统项目

一、登录注册模块 1、注册功能 1.1、注册流程图 1.2、注册代码 /*** 用户注册* param user* return Map<String, Object> 返回错误提示消息&#xff0c;如果返回的 map 为空&#xff0c;则说明注册成功*/public Map<String, Object> register(User user) {Map&l…

YOLOv8实现K折交叉验证教程:解决数据集样本稀少和类别不平衡的难题

本篇博文为大家讲解 YOLOv8 实现 k折交叉验证的流程,包含代码使用和讲解。 K折交叉验证 概念 K折交叉验证 (K-fold cross-validation) 是一种常用于评估机器学习模型性能的技术。 它可以更充分地利用有限的数据集,减少由于数据划分不当而引起的偏差,提高模型的泛化能力。…

计算机里基本硬件的组成以及硬件协同

文章目录 冯诺依曼体系输入设备输出设备存储器运算器控制器协同工作的流程 冯诺依曼体系 世界上第一台通用计算机&#xff0c;ENIAC&#xff0c;于1946年诞生于美国一所大学。 ENIAC研发的前期&#xff0c;需要工作人员根据提前设计好的指令手动接线&#xff0c;以这种方式输入…

卸载大脑,相信DFS

切莫相信动规&#xff0c;吾将为您指明前进之路 印子 比赛时&#xff0c;你是否有这样的经历&#xff1a;不敢用for暴搜&#xff0c;又不会用数学公式推理&#xff1b;焦急地在纸上打草&#xff0c;却没有优化思路&#xff1b;明明比赛前一天晚上背了那么多模板却脑子一片空白…

Modbus tcp转ETHERCAT网关modbus tcp/ip协议

捷米JM-ECT-TCP网关能够连接到Modbus tcp总线和ETHERCAT总线中&#xff0c;实现两种不同协议设备之间的通讯。这个网关能够大大提高工业生产的效率和生产效益&#xff0c;让生产变得更加智能化。捷米JM-ECT-TCP 是自主研发的一款 ETHERCAT 从站功能的通讯网关。该产品主要功能是…

《Deep Leakage from Gradients》

Deep Leakage from Gradients 摘要 现在的分布式机器学习&#xff0c;或联邦学习中最常用的一种方法就是交换梯度&#xff0c;长期以来人们都认为交换梯度的安全的&#xff0c;但是本文提出从公开共享的梯度中获得私有训练数据是可能的。文章将这种泄漏命名为Deep leak from …

PDPS教程:导出带颜色的JT格式2D布局图文件的另一种方法

目录 概述 布局图平面化与边缘化处理 导出为JT格式文件 JT格式文件查看 概述 在PDPS软件中导入机器人生产线或工作站2D布局图之前&#xff0c;要先将布局图文件转换为JT格式。 使用Crossmanager软件转换出来的2D布局图文件会将图形元素原有的颜色自动修改为黑白颜色&…

源码学习初章-基础知识储备

文章目录 学前准备源码地址引言extern "C" 宏定义平台宏跨平台宏vstdio平台禁用警告宏 连接、双层宏定义函数宏系统函数宏自定义函数宏多语句执行宏do while0 普通宏定义 C的一些必备函数知识回调函数和函数指针回调函数wireshark-4.0.7源码例子函数指针wireshark4.0…