美文网首页
验证pytorch是否安装成功(linux)

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

作者: Y_166d | 来源:发表于2019-04-15 16:00 被阅读0次

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

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

sudo lshw -c video

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

nvcc -V

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

python --version

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

import torch
import torchvision

不报错就表明安装成功。

相关文章

网友评论

      本文标题:验证pytorch是否安装成功(linux)

      本文链接:https://www.haomeiwen.com/subject/vzeuwqtx.html