- 系统概述
- ubuntu1604
- cuda10
- cudnn7.5
- 安装pytorch-gpu
- 下载pytorch-gpu版本,下载地址:https://pytorch.org/
- 选择“Stable1.2->Linux->pip->python3.5->cuda10”。如下图所示:
说明:下载前先确认自己的系统、python版本、cuda版本,必须一一匹配。否则安装会出现各种问题。
我下载的版本是:torch-1.2.0-cp35-cp35m-manylinux1_x86_64.whl
- 在终端输入pip3 install torch torchvision 进行安装。
说明:安装过程如果有报错,可以自行百度,百度上可以搜出解决方法。这里不再一一举例了。
pytorch安装成功的标志如下图所示: