简单易懂！Pytorch安装教程（超详细）

已于 2024-01-25 20:18:13 修改
阅读量7.8k 收藏 34
点赞数 28
文章标签： pytorch 人工智能 python 机器学习 深度学习
于 2023-12-21 09:41:44 首次发布
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/2301_81888214/article/details/135123108
版权

在正式开始学习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安装成功。

   在线教程

有需要的小伙伴，可以点击下方链接免费领取或者V扫描下方二维码免费领取🆓
​

请添加图片描述

人工智能书籍

第一阶段：零基础入门（3-6个月）

新手应首先通过少而精的学习，看到全景图，建立大局观。 通过完成小实验，建立信心，才能避免“从入门到放弃”的尴尬。因此，第一阶段只推荐4本最必要的书（而且这些书到了第二、三阶段也能继续用），入门以后，在后续学习中再“哪里不会补哪里”即可。

第二阶段：基础进阶（3-6个月）

熟读《机器学习算法的数学解析与Python实现》并动手实践后，你已经对机器学习有了基本的了解，不再是小白了。这时可以开始触类旁通，学习热门技术，加强实践水平。在深入学习的同时，也可以探索自己感兴趣的方向，为求职面试打好基础。

第三阶段：工作应用

这一阶段你已经不再需要引导，只需要一些推荐书目。如果你从入门时就确认了未来的工作方向，可以在第二阶段就提前阅读相关入门书籍（对应“商业落地五大方向”中的前两本），然后再“哪里不会补哪里”。

 有需要的小伙伴，可以点击下方链接免费领取或者V扫描下方二维码免费领取🆓
​

在这里插入图片描述

人工智能小豪
关注
  • 28
    点赞
  • 踩
  • 34
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
pytorch超级详细的安装教程
于笨笨的博客
05-04 9万+
pytorch超级详细的安装教程 第一次安装的时候安装了差不多有一整天，还安装失败了。这次安装的速度很快。 首先是Anaconda的下载，我用的是上学期学python的时候老师之前给的安装包，很快就安装好了，或者去官网去下载Anaconda也挺快，链接如下： Anaconda官网 其次是Anaconda安装，整个下载过程都是一路next，我只是把下载的路径改变了一下。默认路径是C盘，我调整到了其他盘。 安装完整后，在看是菜单处看到新添加的东西 打开这个Anaconda Prompt应用进入之后是以（bas
深度学习环境搭建pytorch安装教程
11-04
深度学习环境搭建pytorch+CUDA+python3.8
Pytorch最全安装教程（一步到位）
热门推荐
weixin_44904136的博客
03-05 29万+
概述 PyTorch是一个开源的Python机器学习库，其前身是著名的机器学习库Torch。它是美国Facebook公司使用python语言开发的一个深度学习的框架，2017年1月，Facebook人工智能研究院（FAIR）在GitHub上开源了PyTorch。它的前身是2002年诞生于纽约大学 的Torch。 特点： 1）简洁： PyTorch的设计追求最少的封装，尽量避免重复造轮子。 简洁的设计带来的另外一个好处就是代码易于理解。PyTorch的源码只有TensorFlow的十分之一左右，更少的抽
安装Pytorch——CPU版本
qlkaicx的博客
03-16 1768
在Python中，PyTorch库的重要性取决于你的项目需求和个人兴趣。PyTorch是一个流行的深度学习框架，它提供了易于使用的API和强大的功能，使得构建和训练神经网络变得相对简单。因此，如果你打算从事深度学习或机器学习相关的项目，安装PyTorch库是很有必要的。关于是否安装GPU版本的PyTorch，这取决于你的硬件条件和项目需求。GPU版本的PyTorch能够利用图形处理单元（GPU）的并行计算能力来加速神经网络的训练和推理过程。
【超详细教程】2024最新Pytorch安装教程（同时讲解安装CPU和GPU版本）
Little_Carter的博客
01-31 8632
PyTorch是一个开源的深度学习框架，由Facebook人工智能研究院（FAIR）开发。它提供了一套动态张量计算库，具有易于使用、高效性能和强大的扩展性等特点。PyTorch支持在GPU上运行，能够大大加速深度学习模型的训练和推理过程。
pytorch的安装（非常详细）
Redamancy06的博客
07-16 3万+
conda activate pytorch、推荐清华源、中科大源。conda deactivate、查看环境信息 conda info -e、import torch、这个是验证cuda是不是可以正常运行torch.cuda.is_available()、卸载pytorch环境conda remove -n pytorch --all将pycharm的环境换成Anaconda下的pytorch环境......
pytorch环境安装
04-26
pytorch环境安装的PPT，步骤明确，适用于windows各个系统。
PyTorch安装
dkkfff的博客
05-26 76
在 设备管理器-系统设备-硬件-显示 中找到是否有独立显卡（NVIDIA或者AMD），这里是NVIDIA，然后进入 NVIDIA控制面板-帮助-系统-组件 查看可支持的CUDA版本，在官网上去下载该版本以及低版本的都可以，然后根据Pytorch安装教程 根据你的操作系统，第三方库，编程语言，搭载平台可以在官网链接生成不同的安装指令。 ...
2024最新pytorch安装（超详细版）.zip
03-09
### 内容概要 本博客为初学者提供了一个关于如何安装和使用2024年最新版本的PyTorch的详细教程。从PyTorch的介绍开始，逐步介绍了安装PyTorch前的准备工作，包括安装Python和CUDA，然后详细介绍了如何使用pip和conda安装PyTorch。博客还提供了如何验证PyTorch安装是否成功的方法，以及PyTorch的一些高级特性，如动态计算图、GPU加速、丰富的模型库和与Python的集成。最后，博客提供了一些解决安装和使用过程中可能遇到问题的方法。 ### 适用人群 本博客适合对机器学习和PyTorch感兴趣的初学者。无论你是编程小白，还是已经有一定编程基础的读者，都可以从本博客中找到适合自己的学习内容。 ### 使用场景及目标 本博客适用于在家、学校或任何学习环境中自学PyTorch安装和使用。通过跟随博客，读者可以学会如何安装PyTorch，并了解其基本特性和使用方法。博客的目标是帮助初学者建立信心，激发他们对机器学习的兴趣，并为未来的深入学习打下坚实的基础。 ### 其他说明 本博客注重实用性和易懂性，尽量避免使用复杂的专业术语。博客中包含的示例代码和项
pytorch安装教程gpu详细教程.pdf
最新发布
03-27
PyTorch是一个开源的Python机器学习库，基于Torch框架，将Lua语言转译成了Python语言，因此它继承了Torch的灵活性和高效性，并提供了Python API，使得用户可以更方便地...如需更详细的信息和教程，建议查阅PyTorch的官
超简单！pytorch入门教程（五）：训练和测试CNN-附件资源
03-05
超简单！pytorch入门教程（五）：训练和测试CNN-附件资源
简单易懂的PyTorch教程
01-26
pytorch教程 该资源仅供学习 该资源仅供学习 该资源仅供学习 该资源仅供学习
超详细的的PyTorch安装教程，成功率高，适合初学者，亲测可用。
01-31 2万+
Anaconda安装配置PyTorch深度学习教程，简单好用。
Pytorch安装教程（最全最详细版）
MANCXUARY的博客
01-30 4万+
PyTorch是一个开源的深度学习框架，提供了各种张量操作并通过自动求导可以自动进行梯度计算，方便构建各种动态神经网络。 在线安装 1、打开Anaconda Powershell Prompt，输入conda info --envs，查看安装的环境 2、... 离线安装 1、查看Anaconda或者本地安装的Python版本 2、...
2023最新pytorch安装（超详细版）
weixin_44752340的博客
05-16 12万+
因为我之前安装环境花费了不少时间，也是网上搜了很多，想着自己写一个帮助大家快速安装，少走弯路，快速开启深度之路。
安装配置pytorch环境（参考B站“我是土堆”）
weixin_47131505的博客
10-09 8331
安装配置pytorch环境（参考B站“我是土堆”）
pytorch安装教程
06-06
好的，以下是PyTorch的安装教程： 1. 首先需要安装Anaconda，可以在官网下载对应的安装包（https://www.anaconda.com/products/individual）。 2. 安装Anaconda后，在命令行中输入以下命令创建一个新的conda环境，并命名为“pytorch”： ``` conda create --name pytorch python=3.7 ``` 3. 激活刚创建的conda环境： ``` conda activate pytorch ``` 4. 在conda环境中安装PyTorch： ``` conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch ``` 其中，cudatoolkit=10.2表示使用CUDA 10.2版本，如果你的机器上没有安装CUDA可以省略这个参数。 5. 安装完成后，在Python命令行中输入以下代码测试PyTorch是否安装成功： ``` import torch x = torch.rand(5, 3) print(x) ``` 如果输出了一个5行3列的随机矩阵，则说明PyTorch安装成功了。 希望以上内容能够帮助你安装好PyTorch。

“相关推荐”对你有帮助么？

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值