一、pip安装Pytorch
这里已经默认已经安装好了显卡驱动,cuda等。我这里安装的是cuda9与cudnn7,显卡驱动号:Driver Version: 384.130 ;
1、进入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 torch
import torchvision
測試cuda是否可用:
import torch
print(torch.cuda.is_available())
如果没有报错,就证明安装成功!!如果還有問題,請在下面留言