TX2 搭建pytorch v0.3

诗雨影

于 2018-11-20 14:20:23 发布

阅读量1.1k

点赞数

分类专栏：深度学习、TX2、

本文链接：https://blog.youkuaiyun.com/qq_41980584/article/details/84300299

版权

TX2 环境：cuda8.0  cudnn6.0  Ubuntu16.04
参考：https://github.com/dusty-nv/jetson-reinforcement/
$ mkdir pytorch
$ git clone http://github.com/dusty-nv/jetson-reinforcement
$ cd jetson-reinforcement
$ git submodule update --init
$ mkdir build
$ cd build
$ cmake ../
$ make

依次运行以上命令，所需时间较长，要保证网络畅通，所需内存也不小，保证tx2空间够用

验证pytorch是否安装好：

$ python 然后依次输入以下命令，

>>> import torch
>>> print(torch.__version__)
>>> print('CUDA available: ' + str(torch.cuda.is_available()))
>>> a = torch.cuda.FloatTensor(2).zero_()
>>> print('Tensor a = ' + str(a))
>>> b = torch.randn(2).cuda()
>>> print('Tensor b = ' + str(b))
>>> c = a + b
>>> print('Tensor c = ' + str(c))

会出现以下结果：