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

青旅半醒 2023-05-29 11:48 175阅读 0赞

### 一、pip安装Pytorch ###

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

1、进入PyTorch官网：

[pytorch 官网][pytorch]

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

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2NvbXdheV9MaQ_size_16_color_FFFFFF_t_70][]

找到自己所需要下載的版本（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 ###

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

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2NvbXdheV9MaQ_size_16_color_FFFFFF_t_70 1][]

但是，安装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 torch
    
    import torchvision

測試cuda是否可用：

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

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

[pytorch]: https://pytorch.org/
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2NvbXdheV9MaQ_size_16_color_FFFFFF_t_70]: https://img-blog.csdnimg.cn/20191105231529829.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2NvbXdheV9MaQ==,size_16,color_FFFFFF,t_70
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2NvbXdheV9MaQ_size_16_color_FFFFFF_t_70 1]: https://img-blog.csdnimg.cn/20191105233232662.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2NvbXdheV9MaQ==,size_16,color_FFFFFF,t_70