安装GPU版本Pytorch(全网最详细过程)

目录

一、前言

二、安装CUDA

三、安装cuDNN

四、安装Anacanda

五、安装pytorch

六、总结

一、前言

最近因为需要安装GPU版本的Pytorch,所以自己在安装过程中也是想着写一篇博客,把整个过程记录下来,在整个过程中也遇到了不少的问题,查看了以往别人的一些解决方案,希望能够给遇到同样问题的提供一些参考。

如果大家在安装pytorch的时候,并不需要安装gpu版本的,即使用cpu进行训练的pytorch,那么直接在Aanaconda Prompt当中输入以下命令即可:

pip install pytorch

如果是想要安装gpu版本的pytorch,则需要参考以下几个步骤。

二、安装CUDA

以我自己的电脑为例:我的电脑显卡是Geforce MX150,CPU是i5-8250u,操作系统是win10。如果想要使用gpu版本的pytorch,首先得安装一个自己显卡支持的CUDA版本。根据我自己的电脑,选择的是CUDA9.2版本。1、查看电脑的显卡驱动版本:右键选择NVIDIA控制面板→帮助→系统信息

2、查看安装不同版本CUDA所对应的显卡驱动版本(比如这里我选择的就是9.2版本,建议显卡驱动版本高的安装高版本CUDA,显卡驱动版本只要大于CUDA对应的最低版本就行,显卡驱动版本版本过低可以升级):[https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html](https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html "https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html")

 3、接下来下载所选择版本的CUDA,下载网址:[https://developer.nvidia.com/cuda-toolkit-archive](https://developer.nvidia.com/cuda-toolkit-archive "https://developer.nvidia.com/cuda-toolkit-archive")

4、下载完成直接安装即可。

有些人在安装过程中会出现visual studio integration失败的提示,查了一些资料和案例发现最简单的办法就是选择自定义安装,在CUDA组件下取消勾选visual studio integration,这样就不会报错了,另外在安装的时候还出现了弹窗提示脚本错误,这个直接无视就行。

5、安装成功以后可以测试一下,在命令行里面输入nvcc -V,能输出版本则代表安装成功。

三、安装cuDNN

cuDNN需要先注册才能下载:[https://developer.nvidia.com/rdp/cudnn-archive](https://developer.nvidia.com/rdp/cudnn-archive "https://developer.nvidia.com/rdp/cudnn-archive"),进去慢是常态!1、注册完以后选择对应CUDA版本进行下载,我这里选择的是CUDA9.2的对应版本。

2、下载下来以后,解压之后的文件结构如下。

3、接下来把这四个文件全部复制到CUDA的安装文件夹下面。CUDA安装的默认文件夹是:C:\\Program Files\\NVIDIA GPU Computing Toolkit\\CUDA\\v9.2复制完成以后如下图所示:

4、然后测试cuDNN安装是否成功。在extras/demo\_suite/目录下右键在windows终端中打开,然后./bandwidthTest.exe运行,,如果Result = PASS则代表cuDNN安装成功。

四、安装Anacanda

由于我的电脑已经安装过Anaconda,所以此步骤就不再重复,如果不会安装的话,可以去我的另一篇博客里查看,里面有Anaconda官网,里面可以进行下载:[https://blog.csdn.net/Starinfo/article/details/125828332?spm=1001.2014.3001.5501](https://blog.csdn.net/Starinfo/article/details/125828332?spm=1001.2014.3001.5501 "https://blog.csdn.net/Starinfo/article/details/125828332?spm=1001.2014.3001.5501")注意安装时勾选将Anaconda加入到环境变量中。

1、 默认大家已经安装好的情况,打开Anaconda Prompt,通过conda创建一个名为pytorch的虚拟环境,名字可以任取:

conda create -n pytorch python=3.7

2、创建好的虚拟环境可以通过如下命令查看:

conda info --envs

3、然后激活:

conda activate pytorch

五、安装pytorch

1、在清华源镜像中下载pytorch、torchvision、cuda92:

https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/

因为我的压缩包分散了,方便大家看,使用一张别的博主的图:

2、一定要选择正确的版本,带有cpu的版本不要选择,这里选择的pytorch是1.4.0版本,torchvision是0.5.0版本,将上述压缩包复制到Anaconda安装文件夹下对应的pkgs文件夹下。

3、在第三步创建的新环境pytorch中安装下载好的压缩包

conda install D:\Anaconda\pkgs\cuda92-1.0-0.tar.bz2
conda install D:\Anaconda\pkgs\pytorch-1.4.0-py3.7_cuda92_cudnn7_0.tar.bz2
conda install D:\Anaconda\pkgs\torchvision-0.5.0-py37_cu92.tar.bz2

安装完以后如图所示:

4、到这一步,该环境已经搭建完成了,接下来将该环境加入到pycharm中,在file→Settings→Python Interpreter中点击右侧的齿轮,选择add->Existing environment,选择刚才创建好的pytorch环境,这里因为我已经创建好了,忘了截图,所以用一张别的博主的图给大家作演示:

5、验证是否安装正确,可以运行下面代码,看是否得到输出。
import torch
import numpy as np
test=np.array([[1,2,3,4],[5,6,7,8]])
x=torch.tensor(test)
print(x)import torch
print(torch.cuda.is_available())  #输出为True,则安装无误

六、总结

至此所有的工作都已经完成,接下来就可以在pycharm中进行pytorch的GPU版本的学习和训练。本篇博客是跟据我个人的安装情况和其他博主的安装情况综合而成,有什么遗漏或错误的的地方欢迎大家指正!

#学习资源推荐

零基础Python学习资源介绍

👉Python学习路线汇总👈
Python所有方向的技术点做的整理,形成各个领域的知识点汇总,它的用处就在于,你可以按照上面的知识点去找对应的学习资源,保证自己学得较为全面。(学习教程文末领取哈)
在这里插入图片描述

👉Python必备开发工具👈
在这里插入图片描述

温馨提示:篇幅有限,已打包文件夹,获取方式在:文末

👉Python学习视频600合集👈
观看零基础学习视频,看视频学习是最快捷也是最有效果的方式,跟着视频中老师的思路,从基础到深入,还是很容易入门的。
在这里插入图片描述

👉实战案例👈
光学理论是没用的,要学会跟着一起敲,要动手实操,才能将自己的所学运用到实际当中去,这时候可以搞点实战案例来学习。
在这里插入图片描述

👉100道Python练习题👈
检查学习结果。
在这里插入图片描述
👉面试刷题👈
在这里插入图片描述在这里插入图片描述
在这里插入图片描述
在这里插入图片描述

资料领取

上述这份完整版的Python全套学习资料已经上传CSDN官方,朋友们如果需要可以微信扫描下方CSDN官方认证二维码输入“领取资料” 即可领取。

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

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

相关文章

Vite 官方文档速通

前言:参考 Vite 官网 一. 指引 1. 为什么选择 Vite 1.1 现实问题 问题:当项目中 JS 代码量变多,就会导致:1. 启动开发服务慢(几分钟)。2,模块热替换(HMR)慢&#xff0…

Java List中对象根据id去重,并处理重复对象的某个字段

List中对象根据id去重 一、需求二、解决 一、需求 参考文章:https://blog.csdn.net/A_Gui_Code/article/details/106978867 对在list集合中对象根据主键id去重,同时需要对重复对象的某个字段进行单独处理。 例如,对象包含字段如下, 当某个对象重复时&a…

nignx日志定时清理、Windows 清理nignx日志、linux 清理nignx日志

因为项目日志老占磁盘空间,记录一下优化配置文件: 针对linux清理: 1.创建一个名为 cleanup_nginx_logs.sh 的脚本文件 nano cleanup_nginx_logs.sh 2.在打开的文本编辑器中,将以下内容复制并粘贴到文件中: #!/bin…

【开源】基于JAVA+Vue+SpringBoot的用户画像活动推荐系统

目录 一、摘要1.1 项目介绍1.2 项目录屏 二、功能模块2.1 数据中心模块2.2 兴趣标签模块2.3 活动档案模块2.4 活动报名模块2.5 活动留言模块 三、系统设计3.1 用例设计3.2 业务流程设计3.3 数据流程设计3.4 E-R图设计 四、系统展示五、核心代码5.1 查询兴趣标签5.2 查询活动推荐…

状态压缩 笔记

棋盘式的f[i][j]中表示状态的j可以是状态本身也可以是在合法状态state中的下标 用状态本身比较方便,用下标比较省空间 用下标的话可以开id[M]数组记录一下 蒙德里安的梦想 求把 NM的棋盘分割成若干个 12的长方形,有多少种方案。 例如当 N2&#xff0…

【Redis】一文搞懂redis的所有知识点

目录 1. 什么是Redis?它主要用来什么的? 2.说说Redis的基本数据结构类型 2.1 Redis 的五种基本数据类型​编辑 2.2 Redis 的三种特殊数据类型 3. Redis为什么这么快?​编辑 3.1 基于内存存储实现 3.2 高效的数据结构 3.3 合理的数据编…

新火种AI|哄哄大模型的火爆,给了普通人AI创业破局的关键

作者:一号 编辑:美美 人们似乎更喜欢把AI当做玩具,而非工具。 近日,一款名为哄哄模拟器的AI原生应用火了,一天之内就吸引了60万用户。 哄哄模拟器设置了多种情侣吵架场景,无论你是男是女,都…

光伏设计系统都具备哪些功能?

随着可再生能源的日益重要,光伏能源已成为我们能源结构中的重要组成部分。而光伏设计系统作为实现光伏能源高效利用的关键,其功能也日益丰富和多样化。本文将探讨光伏设计系统所具备的主要功能。 1.数据分析与模拟 光伏设计系统能够对大量的数据进行分…

【经典项目】Java小游戏 —— 贪吃蛇

一、需求分析 当提到贪吃蛇游戏时,通常是指一款经典的电子游戏,玩家通过操作一个蛇在屏幕上移动,目标是吃掉各种食物并尽量避免碰撞到自己的身体或游戏区域的边界。 贪吃蛇游戏最早出现在20世纪70年代末的个人电脑上,后来在各种游…

【CSS】动画(Transform,Transition,Animation)

CSS3属性中关于制作动画的三个属性:Transform,Transition,Animation。 transform 描述了元素的静态样式,本身不会呈现动画效果,可以对元素进行旋转rotate、扭曲skew、缩放scale和移动translate以及矩阵变形matrix。 div{transform:scale(2…

行为型设计模式—访问者模式

访问者模式:将数据结构对象与数据操作分离的设计模式,可以在不改变数据结构对象类结构的前提下定义作用于这些对象的新的操作。 访问模式结构: 访客接口 (Visitor) 声明了一系列以表示对象结构的具体元素为参数的访问…

使用 Paimon + StarRocks 极速批流一体湖仓分析

摘要:本文整理自阿里云智能高级开发工程师王日宇,在 Flink Forward Asia 2023 流式湖仓(二)专场的分享。本篇内容主要分为以下四部分: StarRocksPaimon 湖仓分析的发展历程使用 StarRocksPaimon 进行湖仓分析主要场景和…

ONLYOFFICE是一站式协作免费开源办公神器

一、前言 众所周知,我们工作都离不开办公软件,一个好的办公软件可以大幅度提高工作效率,就目前市场而言,常见的办公软件有微软的 Microsoft Office 以及国产的 WPS Office,微软的 Microsoft Office是付费的&#xff0…

在JAVA中如何删除ZIP等等的压缩文件

删除ZIP文件的步骤和要点 JAVA中文件的类型在JAVA中能够被删除的文件类型问题:既然如此,那为什么有时候会遇到文件无法被删除的问题?心得总结 JAVA中文件的类型 在Java中,只存在两种文件类型,一种是文件夹&#xff0c…

postgresql和kingbase关于模糊查询大小写兼容问题

在mysql中,会有相关的like关键词,并且默认的是忽略大小写的。但是在postgresql和kingbase中,只有ilike关键字,并且默认是大小写敏感的。当我们使用mybatisplus的时候,默认提供的api也只有like()。这里提供一种方式来对…

CANoe实际项目中文件夹的规划

本人,之前设计了一个CANoe工程,由于工程设计之初没有设计好文档的归纳分类,导致文件查找起来非常费劲。 为了避免以后出现文件混乱,不可查找的问题,故特此归纳说明。 建立工程时: 第1步就应该设计好文档…

基于51单片机的加油站计费系统

基于51单片机的加油站计费系统[proteus仿真] 计费检测系统这个题目算是课程设计和毕业设计中常见的题目了,本期是一个108基于51单片机的加油站计费系统 需要的源文件和程序的小伙伴可以关注公众号【阿目分享嵌入式】,赞赏任意文章 2¥&#…

“软件老兵”的新征程:从代码到方案,40岁转身更显智慧锋芒

在快速迭代的软件行业,许多经验丰富的软件工程师,随着年岁的增长,发现自己在纯技术领域的竞争力似乎逐渐不敌那些刚刚踏入职场的00后工程师。 面对这一挑战,一些“软件老兵”选择转换跑道,将目光投向了方案工程师这一新…

EasyExcel实现三级联动

前言 项目中需要在导出的模板中新增三级联动的功能,类似省市区的联动。在网上找了一些方法,都不能直接使用,需要进行修改。本文主要分享一下,改后的代码,可以直接使用。 代码 public class CascadeWriteHandler imp…

Linux同时安装多个版本的JDKlinux多版本JDK切换配置- [Centos 安装JDK]

Linux同时安装多个版本的JDK&linux多版本JDK切换配置- [Centos 安装JDK] 一、查询系统是否已经安装jdk rpm -qa|grep java 或 rpm -qa|grep jdk 或 rpm -qa|grep gcj 二、卸载已安装的jdk Centos7存在已安装的JDK1.7时,可以使用以下方式删除1.7的openJDK rp…