目录

简介

下面介绍ubuntu下的pytorch安装。

安装pytorch

conda install pytorch torchvision cuda80 -c soumith

显示显卡信息

sudo lshw -c video

显示CUDA信息

nvcc -V

查看Phthon版本

python --version

验证

验证是否安装成功

 python
 import torch
 import torchvision