PyTorch 学习之Ubuntu16.04 下安装 PyTorch（conda与pip两种方式安装）

2024-05-30 18:04:35

一、pip安装Pytorch

这里已经默认已经安装好了显卡驱动，cuda等。我这里安装的是cuda9与cudnn7，显卡驱动号：Driver Version: 384.130 ；

1、进入PyTorch官网：

pytorch 官网

进去后，点击左下角 previous versions of PyTorch

找到自己所需要下載的版本（torch 与torchvison）.如系統,cuda版本,python版本。

进入PyTorch的下载目录，使用pip3命令安装：（python3是用pip3，python2就直接用pip，根据自己的情况，注意这点不同）

sudo pip3 install torch-0.3.1-cp35-cp35m-linux_x86_64.whl

torch-0.3.1-cp35-cp35m-linux_x86_64.whl爲你下載的文件。
安装完PyTorch后，要安装torchvision，方法如下：

目前我了解的torc与torchvision版本的匹配：

torch 1.1 torchvision0.3

torch 1.2 torchvision0.4

torch 1.0一下的 torch0.2或者0.1

依然在下载torch那的链接下载torchvision，同样的安装方式；

sudo pip3 install torchvision-版本号.whl

卸载：

sudo pip3 uninstall torchvision

二、conda安裝Pytroch

按照正常安裝，是下圖所示，直接使用命令進行安裝；

但是，安装pytorch官方指定的方法conda下载torch太慢。因此交給大家一个方法，

首先添加一個清華源來加快速度；

解决步骤：

第一步：进入添加清華源：

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes

然後添加pytroch源：

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

最後安裝：

conda install pytorch torchvision cudatoolkit=9.0（这个时候要注意，我们已经吧-c pytorch给删除了）

解释一下，就是我们没有使用pytorch的源，而改成清华镜像的源了；

三、檢驗是否安裝成功

安装成功PyTorch和torchvision后，在终端输入：

python3

进入python3的环境

然后输入：

import torchimport torchvision

測試cuda是否可用：

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

如果没有报错，就证明安装成功！！如果還有問題，請在下面留言

PyTorch 学习之Ubuntu16.04 下安装 PyTorch（conda与pip两种方式安装）相关推荐

【Linux】17.Ubuntu16.04 系统添加网关和DNS的两种方式
注意,Ubuntu18.04和Ubuntu16.04 系统有很大不同,所以以下方法只适用于Ubuntu16.04 . Ubuntu16.04 系统添加网关和DNS的方式 1.下面这种临时添加网关的方式 ...
centos6配置日志外发_CentOS6下记录后台操作日志的两种方式
CentOS6下记录后台操作日志的两种方式平时为了记录登录CentOS Linux系统的操作命令,需要将操作日志记录下来,下面介绍两种方式 1.利用script以及scriptreplay工具 sc ...
实现Windows下Qt扫描U盘的两种方式
参考网上资源,总结了如下实现Windows下Qt扫描U盘的两种方式:. 方式1: 检测Windows的事件–WM_DEVICECHANGE 缺点:想要检测时,必须发生过一次U盘插拔动作,需要手动&qu ...
centos 6.8安装git_搭建一个开源项目3两种方式安装git
一.开始工具的安装 1.git 安装git工具有两种方式,一种就是利用自带包管理工具,一种是源码编译安装 (1)由于CentOS已经具有包管理器因此只需要一行命令即可自动安装 yum install ...
Ubuntu 安装 JDK 7 / JDK8 的两种方式
2019独角兽企业重金招聘Python工程师标准>>> ubuntu 安装jdk 的两种方式: 1:通过ppa(源) 方式安装. 2:通过官网下载安装包安装. 这里推荐第1种,因为可 ...
ubuntu16.04 Linux+xfce4的文件夹出现两种背景颜色+文件夹中右键打开终端失灵了.
好不容易把xfce4的文件夹弄成了夜间模式,结果用着用着发现文件夹有两种颜色可以看到一种豆沙绿,一种是黑色的. 这个问题困扰我好几个月了..总是用着用着就发现居然变成了两种颜色. 还有更稀奇古怪的, ...
Linux系统-安装显卡GPU驱动的两种方式
方案1 1)先到NVIDIA的官方下载 xx.run GPU驱动文件,可以到官方下载对应的版本: https://www.nvidia.cn/Download/Find.aspx?lang=cn ub ...
快速上手Appium（一）-两种方式安装Appium
前言大部分移动端自动化测试工程师刚开始接触的移动端UI自动化测试框架都是Appium,它是一系列很优秀的开源软件,适合用于做移动端UI自动化测试或者远程控制,本篇文章是Appium快速入门系列的第一 ...
Andorid开发学习---ubuntu 12.04下搭建超好用的安卓模拟器genymotion 安装卸载virtualbox 4.3...
什么是Genymotion? Genymotion是一套完整的工具,它提供了Android虚拟环境.它简直就是开发者.测试人员.推销者甚至是游戏玩家的福音. Genymotion支持Windows.L ...

最新文章

热门文章