1、新买的TX2建议直接刷机,不要用自带系统,刷机时安装jetpack中所有包
2、刷机:当安装完系统镜像时,会提示你重启,先重启,再安装其他的包(cuda、cudnn等)(注意:将系统镜像那一项以上的都选为no action,只安装target board)
2、刷机:当安装完系统镜像时,会提示你重启,先重启,再安装其他的包(cuda、cudnn等)(注意:将系统镜像那一项以上的都选为no action,只安装target board)
3、等待安装完成后,开始安装tensorflow,采用编译安装,详见链接
https://syed-ahmed.gitbooks.io/nvidia-jetson-tx2-recipes/content/first-question.html 这个链接已实验成功,不过tf版本为1.0
http://www.jetsonhacks.com/2017/09/14/build-tensorflow-on-nvidia-jetson-tx2-development-kit/ 这个链接也实验过,版本为1.3,未成功调用gpu
4、安装jupyter notebook推荐
注意事项:
1、编译前增加swap空间为8G
2、测试程序记得配置session:
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True,allow_soft_placement=True))
之前一直没设置 allow_soft_placement参数,导致没有成功使用gpu
根据官方doc说明,这个是当指定设备不可用时寻找替代设备的标志位,至于为什么这样就可以了,还不太清楚。