search

peterlee0127 / tensorflow-nvJetson

TensorFlow for NVIDIA Jetson, also include patch and script for building.
https://tfjetson.peterlee.app
205 stars 61 forks source link

TensorFlow for JetPack3.2 #7

Closed AastaNV closed 6 years ago

AastaNV commented 6 years ago

Hi, peterlee0127

This is Aasta from NVIDIA.

Thanks for sharing your TensorFlow build for JetPack3.1. It really helps our user.

Could you also update a TensorFlow wheel for JetPack3.2 DP, which uses CUDA 9.0/cuDNN 7? The upgrade will be extremely helpful to us.

Thanks.

LaurensVanAcker commented 6 years ago

+1

peterlee0127 commented 6 years ago

I just update JetPack 3.2 today. The tensorflow 1.5 will suupport cuda 9 and cudnn 7. I build tensorflow 1.5rc with Cuda 9. It still need time to modify some code.

Anyway, if I build tensorflow with cuda 9 successful. I will update it.