简单易懂!Pytorch安装教程(超详细)

在正式开始学习Pytorch之前,安装Pytorch同样是重要的一个环节。我将安装Pytorch的主要过程以及遇到的一些问题写在下面,希望能对各位有所帮助。

一、系统与环境说明

在开始用Pytorch进行深度学习之前,要先准备好基本的软硬件环境。下面我分别从操作系统、GPU环境等方面简单说一下最基本的Pytorch运行环境。

1、操作系统

目前Pytorch支持的操作系统有Windows、Linux、MacOS。那么究竟使用哪一种操作系统呢?

如果您只是简单的使用Pytorch,不需要CUDA编程或者使用高度定制化的操作。那么任意一种操作系统都可以。

如果您需要高度定制化的操作,比如说定义一个CUDA函数,那么个人建议使用Linux或者Mac。

如果您已经装了Windows并且不想卸载重装系统,那么您可以安装一个双系统。安装双系统前切记切记要把系统备份做好,因为手滑格式化硬盘的惨剧也不再少数。

2、CPU与GPU

如果您想深入学习Pytorch或者Tensorflow,那么一块GPU是少不了的。没有GPU许多实验都无法迅速完成。如果没有GPU,就请使用CPU吧!把数据集缩小,过把瘾总是可以的。

一些朋友也许会问,那么能不能用带GPU的笔记本搞深度学习呢?当然可以!只要程序运行的时间不是太长,用带GPU的笔记本也能完成任务。游戏都能玩,程序还跑不了吗?不过如果有条件,我建议还是买一台台式机,再根据经济条件配一块显卡(二手的显卡其实也能用)。毕竟笔记本散热差,而且笔记本上的显卡很多时候是被阉割过的,性能比较差。当然,如果只是用于学习目的,一个笔记本电脑已经足够了。

GPU也有许多型号,如果要用Pytorch请准备1050以及以上的显卡。10系列以下的显卡上是运行不了最新版本的Pytorch的。

二、安装驱动和CUDA环境

下面的教程将介绍如何在Ubuntu18上安装Pytorch。安装过程非常简单,其中最大的困难在于网速太慢。

首先英伟达官网下载CUDA以及cudnn加速包,这里的下载cudnn需要注册英伟达账号。

1、下载安装CUDA

访问英伟达官网下载CUDA,链接为https://developer.nvidia.com/cuda-downloads。注意请下载runfile,这个比较稳妥。CUDA版本请选择10.1.如果您还想装tensorflow,那么请您装CUDA10.0。

下载安装指令如下所示:

#下载 
wget http://developer.download.nvidia.com/compute/cuda/10.1/Prod/local_installers/cuda_10.1.243_418.87.00_linux.run 
#安装 
sudo sh cuda_10.1.243_418.87.00_linux.run 

许多教程里说先装驱动再安装CUDA。而CUDA里自带的驱动其实也能用(只要显卡是10+以上的版本),索性一并安装了。

敲完上面代码后会跳出来一个readme。请按住空格。不要按回车键。直到出现下面的内容。然后会有一个选择安装内容的选项。“+”号就是安装。“-”号就是不安装。全部安装吧!然后会出现类似下面的内容。我这里列举的是CUDA9.0的内容。CUDA10.1与之类似。

Do you accept the previously read EULA?  
accept/decline/quit: accept  
Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 384.81?  
(y)es/(n)o/(q)uit: y  
Install the CUDA 9.0 Toolkit?  (y)es/(n)o/(q)uit: y  
Enter Toolkit Location  [ default is /usr/local/cuda-9.0 ]:   
Do you want to install a symbolic link at /usr/local/cuda?  
(y)es/(n)o/(q)uit: y  
Install the CUDA 9.0 Samples?  
(y)es/(n)o/(q)uit: y  
Enter CUDA Samples Location  [ default is /home/pertor ]:   
Installing the CUDA Toolkit in /usr/local/cuda-9.0 ...  
Missing recommended library: libXmu.so 

装完了之后需要配置环境变量。

切记!!!要配置环境变量。否则系统找不到相关路径。打开主目录下的 .bashrc文件添加如下路径,具体命令如下所示。

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-10.1/lib64 
export PATH=$PATH:/usr/local/cuda-10.1/bin 
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-10.1 

终端运行:source ~/.bashrc

检查:nvcc --version,如果显示nvcc版本就说明安装成功了。

2、安装cudnn

cudnn就是一个神经网络加速包,也谈不上安装,把包复制进去就好了。

(1)下载安装文件

按需求下载cudnn的安装文件:https://developer.nvidia.com/rdp/cudnn-archive

先选中10.1版本,挑第三个cuDNN Library for Linux下载。

(2)安装cudnn

解压下载的文件,可以看到cuda文件夹,在当前目录打开终端,执行如下命令:

sudo cp cuda/include/cudnn.h /usr/local/cuda/include/ 
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/ 
sudo chmod a+r /usr/local/cuda/include/cudnn.h 
sudo chmod a+r /usr/local/cuda/lib64/libcudnn* 

这样,CUDA环境就安装成功了。

三、安装Pytorch以及相关包

下面要到Pytorch官网下载相应的安装包。现在的稳定版为1.3版本。直接用pip进行安装吧!

命令如下:

sudo pip3 install torch torchvision 

当然,网速肯定是非常差的。怎么办呢?可以设置临时的Pip源

命令如下:

sudo pip3 install torch torchvision -i https://pypi.mirrors.ustc.edu.cn/simple/ 

你会发现运行如飞。

下面要对是否安装成功进行测试。

首先在终端敲一下:

python3 

进入python命令行后,执行以下命令。如果不报错就说明安装成功了。

import torch 
x = torch.empty(5, 3) 
print(x) 

如果报错,十之八九是环境变量没配置好。请在下方留言,把相关的错误代码贴上来。我们一起分析分析。

至此Pytorch安装成功。

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

如若内容造成侵权/违法违规/事实不符,请联系我们进行投诉反馈qq邮箱809451989@qq.com,一经查实,立即删除!

相关文章

深入理解网络 I/O:FileOutputStream、BufferFileOutputStream、ByteBuffer

🔭 嗨,您好 👋 我是 vnjohn,在互联网企业担任 Java 开发,CSDN 优质创作者 📖 推荐专栏:Spring、MySQL、Nacos、Java,后续其他专栏会持续优化更新迭代 🌲文章所在专栏&…

怎么选择高压放大器(高压放大器选型指南)

在许多科学、工程和实验应用中,需要对高压信号进行放大,以便进行测量、激励或其他各种目的。选择适当的高压放大器对于系统性能至关重要。下面将带来高压放大器选型指南的介绍,帮助工程师们在众多选项中做出明智的选择。 1.确定应用需求 首先…

规则引擎调研情况 URule Pro、VRS和ILOG ODM

最近调研了三家公司的规则引擎产品,各有利弊,具体情况如下: 1. URule Pro 配置本地环境: Web端测试样例: 产品特点: 编程语言 Java语言 是否有Python接口 否 核心算法 3.0及之前的版本 Rete算法…

应用 Strangler 模式将遗留系统分解为微服务

许多来源在一般情况下提供了微服务的解释,但缺乏特定领域的示例。新来者或不确定从哪里开始的人可能会发现掌握如何将遗留系统过渡到微服务架构具有挑战性。本指南主要面向那些正在努力启动迁移工作的个人,它提供了特定于业务的示例来帮助理解该过程。 …

还在用nvm?来试试更快的node版本管理工具——fnm

前言 📫 大家好,我是南木元元,热衷分享有趣实用的文章,希望大家多多支持,一起进步! 🍅 个人主页:南木元元 目录 什么是node版本管理 常见的node版本管理工具 fnm是什么 安装fnm …

IDEA 设置 SpringBoot logback 彩色日志(附配置文件)

1、背景说明 最开始使用 SpringBoot 时,控制台日志是带彩色的,让人眼前一亮😄 后来彩色莫名丢失,由于影响不大,一直没有处理。 2、配置彩色 最近找到了解决方法(其实是因为自定义 logback.xml&#xff0…

多媒体互动橱窗设计如何改变内容展示形式?

橱窗设计在展品展示中扮演着举足轻重的角色,它相较于传统展示形式,能更直观地呈现展品效果,而且优质的橱窗设计还能提升品牌的产品形象,正因此,也被广泛应用于企业、博物馆、店铺等场所。随着多媒体技术的蓬勃发展和行…

【MATLAB第84期】基于MATLAB的波形叠加极限学习机SW-ELM代理模型的sobol全局敏感性分析法应用

【MATLAB第84期】基于MATLAB的波形叠加极限学习机SW-ELM代理模型的sobol全局敏感性分析法应用 前言 跟往期sobol区别: 1.sobol计算依赖于验证集样本,无需定义变量上下限。 2.SW-ELM自带激活函数,计算具有phi(x)e^x激…

命令执行 [SWPUCTF 2021 新生赛]babyrce

打开题目 我们看到题目说cookie值admin等于1时,才能包含文件 bp修改一下得到 访问rasalghul.php,得到 题目说如果我们get传入一个url且不为空值,就将我们get姿势传入的url的值赋值给ip 然后用正则过滤了 / /,如果ip的值没有 / …

Maven将Jar包打入本地仓库

Maven将Jar包打入本地仓库 Maven将Jar包打入本地仓库嘚吧嘚下载Maven配置Maven新建MAVEN_HOME编辑Path验证Maven配置 Jar包打入Maven仓库 Maven将Jar包打入本地仓库 嘚吧嘚 最近项目用到一个Jar包,不能从远程仓库拉取,只有一个Jar包,所以需…

BSWM 模式管理(二)ESH

BSWM 模式管理 ESH 1 ECU State Handling (ESH)2 BSWM ESH 五大模式与六大通用状态机3 状态机对应的切换条件 conditions or rules4 默认主要的 ACTION 或者 ACTION LIST1 ECU State Handling (ESH) 与 ECUM 相关,整个 ECU 状态管理的状态机制 2 BSWM ESH 五大模式与六大通…

基于 Flink 的典型 ETL 场景实现方案

目录 1.实时数仓的相关概述 1.1 实时数仓产生背景 1.2 实时数仓架构 1.3 传统数仓 vs 实时数仓 2.基于 Flink 实现典型的 ETL 场景 2.1 维表 Join ■ 2.1.1 预加载维表 方案 1: 方案 2: ■ 2.1.2 热存储关联 ■ 2.1.3 广播维表 ■ 2.1.4 Tem…

福建省大数据集团数据应用开发大赛全面升级

更多精彩,请持续关注 点击链接直达大赛官网 福建省大数据集团数据应用开发大赛 (fjbdg.com.cn)

车辆违规开启远光灯检测系统:融合YOLO-MS改进YOLOv8

1.研究背景与意义 项目参考AAAI Association for the Advancement of Artificial Intelligence 研究背景与意义 随着社会的不断发展和交通工具的普及,车辆违规行为成为了一个严重的问题。其中,车辆违规开启远光灯是一种常见的违规行为,给其…

DBeaver中使用外部格式化程序对进行sql格式化

本文介绍了如何在DBeaver中使用外部格式化程序对sql进行格式化。 一、pgFormatter 1.准备工作 下载地址:https://github.com/darold/pgFormatter/releases/ pgFormatter是perl脚本,所以需要perl运行环境支持。 perl下载地址:https://str…

【力扣100】543.二叉树的直径

添加链接描述 # Definition for a binary tree node. # class TreeNode: # def __init__(self, val0, leftNone, rightNone): # self.val val # self.left left # self.right right class Solution:def __init__(self):self.max 0def diamete…

实现linux与windows进行文件共享

目录 一.展现形式 二.场景需求 三.具体操作 1.windows访问sumba 2.Linux访问sumba 一.展现形式 支持以文件夹的形式可视化操作文件 二.场景需求 1.有一台Linux物理机 2.有一台window物理机 3.Linux已配置好sumba服务器 三.具体操作 1.windows访问sumba 首先按下winR…

PyTorch深度学习实战(26)——卷积自编码器(Convolutional Autoencoder)

PyTorch深度学习实战(26)——卷积自编码器 0. 前言1. 卷积自编码器2. 使用 t-SNE 对相似图像进行分组小结系列链接 0. 前言 我们已经学习了自编码器 (AutoEncoder) 的原理,并使用 PyTorch 搭建了全连接自编码器,但我们使用的数据…

【C++初阶】第一站:C++入门基础(下)

前言: 紧接着上两篇文章,c入门基础(上):C入门基础(上) c入门基础(中):C入门基础(中) 继续补充完c初阶入门基础的知识点,本章知识点包括: 引用和指针的区别、内联函数、auto关键字(C11)、基于范围的for循环…

企业在什么场景下使用Windows活动目录?

Windows活动目录是微软提供的一种集中式身份验证和访问控制服务,它具有许多功能和优势,因此在很多企业中被广泛使用。那么,企业在什么场景下会选择使用Windows活动目录呢? 首先,Windows活动目录适用于中大型企业或组织…