验证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
不报错就表明安装成功。