您的位置:首页 > 其它

Jetson TX2刷机及安装tensorflow gpu注意事项

2017-10-03 12:11 991 查看
1、新买的TX2建议直接刷机,不要用自带系统,刷机时安装jetpack中所有包

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说明,这个是当指定设备不可用时寻找替代设备的标志位,至于为什么这样就可以了,还不太清楚。
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  Jetson TX2 tensorflow gpu