Ubuntu16.0.4下安装深度学习框架：PyTorch

最新推荐文章于 2025-02-28 10:30:16 发布

dulingwen

最新推荐文章于 2025-02-28 10:30:16 发布

阅读量1k

点赞数

分类专栏：深度学习

本文链接：https://blog.youkuaiyun.com/dulingwen/article/details/88928362

版权

深度学习专栏收录该内容

16 篇文章

订阅专栏

本文详细介绍了如何在Ubuntu 16.0.4上安装支持CUDA的PyTorch深度学习框架，包括创建环境、安装PyTorch及测试安装是否成功等步骤，并提供了官方教程链接。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

PyTorch 是Facebook于2017年推出的一个深度学习框架，支持采用 Python 语言编程。PyTorch推出之后大受欢迎，现在已经是 GitHub 上增长速度第二快的开源项目，在过去的 12 个月里，贡献者增加了 2.8 倍，仅次于Tensorflow。2018年12月8日官方已经发布 PyTorch1.0 稳定版。笔者在此推荐一本书籍叫做《深度学习入门之PyTorch》，索取地址：https://github.com/strivebo/pytorch-learning/tree/master/Books。

下面讲述一下在ubuntu16.0.4上安装PyTorch（支持CUDA的版本）的过程。

必备条件：python、包管理器（anaconda或pip）。笔者推荐anaconda，可以省去很多繁琐的手动安装过程。

一、创建pytorch环境：

conda create -n pytorch

创建完成后激活环境：

conda activate pytorch

二、安装pytorch

打开网站：https://pytorch.org/get-started/locally/，根据自己的情况选择安装命令，pytorch版本建议选择稳定的1.0版本。

执行图中给出的安装命令：

conda install pytorch torchvision cudatoolkit=9.0 -c pytorch

然后会安装很多工具，由于网络原因其中的pytorch—1.0.1可能安装很慢甚至无法安装，建议多试几次，或者尝试配置清华源、阿里源等快速源。然后等待一段时间就安装成功了。

三、安装显卡驱动

参考我的博客：https://blog.youkuaiyun.com/dulingwen/article/details/88565588

四、测试是否安装成功

1. 为了确保pytorch被正确安装，进入python，输入如下示例代码：

from __future__ import print_function
import torch
x = torch.rand(5, 3)
print(x)

如果输出以下类似内容，说明安装成功：

tensor([[0.3380, 0.3845, 0.3217],
        [0.8337, 0.9050, 0.2650],
        [0.2979, 0.7141, 0.9069],
        [0.1449, 0.1132, 0.1375],
        [0.4675, 0.3947, 0.1426]])

2. 另外，为了检查你的GPU和CUDA能够被PyTorch使用，可以执行下面的代码，若返回True，说明CUDA可以使用：

import torch
torch.cuda.is_available()

输出：

True

pytorch官方教程：https://pytorch.org/tutorials/index.html