验证pytorch是否安装成功(linux)

2019-04-15  本文已影响0人  Y_166d

验证pytorch是否安装成功(linux)

要显示显卡信息,在终端输入:

sudo lshw -c video

要显示CUDA信息,在终端输入:

nvcc -V

查看Phthon版本,在终端输入:

python --version

验证pytorch是否安装成功,在终端进入python环境
此时进入python环境。
然后,继续输入

import torch
import torchvision

不报错就表明安装成功。

上一篇 下一篇

猜你喜欢

热点阅读