一、pip安裝Pytorch

? ? ?這里已經(jīng)默認已經(jīng)安裝好了顯卡驅動，cuda等。我這里安裝的是cuda9與cudnn7，顯卡驅動號：Driver Version: 384.130?；

? ?1、進入PyTorch官網(wǎng)：

? ? ? ? ? ?pytorch 官網(wǎng)

進去后，點擊左下角?previous versions of PyTorch

找到自己所需要下載的版本（torch 與torchvison）.如系統(tǒng),cuda版本,python版本。

進入PyTorch的下載目錄，使用pip3命令安裝：（python3是用pip3，python2就直接用pip，根據(jù)自己的情況，注意這點不同）

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（這個時候要注意，我們已經(jīng)吧-c pytorch給刪除了）

?解釋一下，就是我們沒有使用pytorch的源，而改成清華鏡像的源了；

三、檢驗是否安裝成功

安裝成功PyTorch和torchvision后，在終端輸入：

python3

進入python3的環(huán)境

然后輸入：

import torchimport torchvision

測試cuda是否可用：

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

如果沒有報錯，就證明安裝成功！！如果還有問題，請在下面留言

?

?

總結

以上是生活随笔為你收集整理的PyTorch 学习之Ubuntu16.04 下安装 PyTorch（conda与pip两种方式安装）的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。