Ubuntu16.04安装pytorch方法

准备工作

安装cuda10.0及cudnn7.4.

请参考我这篇文章：Ubuntu16.04安装cuda及cudnn方法

下载pytorch安装文件

torch官网上可找到各种安装方式，这里选择whl安装，首先在网站上下载对应cuda10.0版本的whl文件。这里我下载的是1.0.1.post2(一个项目指定要用这个版本，:<我也没办法)，又由于我用的python3.5，因此选择下载torch-1.0.1.post2-cp35-cp35m-linux_x86_64.whl这个安装。

安装pytorch

sudo chmod a+x torch-1.0.1.post2-cp35-cp35m-linux_x86_64.whl
pip install torch-1.0.1.post2-cp35-cp35m-linux_x86_64.whl

另外还需安装torchvision，这里对应的版本是0.2.2.post2，用pip安装：

pip install torchvision==0.2.2.post2 -i https://pypi.tuna.tsinghua.edu.cn/simple/

测试是否安装成功：

import torch
print(torch.__version__)           #输出Pytorch的版本
print(torch.cuda.is_available())   #看看能否正常调用CUDA

一切正常，大功告成。

本文地址：https://blog.csdn.net/wolfcsharp/article/details/107474382