Skip to content

Commit

Permalink
Add instructions on how to support GPU with TFJava 1.0.0
Browse files Browse the repository at this point in the history
  • Loading branch information
karllessard committed Nov 1, 2024
1 parent d625dbb commit 127013b
Showing 1 changed file with 18 additions and 0 deletions.
18 changes: 18 additions & 0 deletions MIGRATING.md
Original file line number Diff line number Diff line change
Expand Up @@ -49,6 +49,24 @@ The Java Module (jigsaw) names has been updated to drop the leading `org.`, as f
- `tensorflow-core-native` : `tensorflow.nativelib`
- `tensorflow-framework` : `tensorflow.framework` (was `org.tensorflow.framework` before)

### GPU Support

Previous versions of TF Java were building a `tensorflow-core-platform-gpu` artifact upon which application could depend
on to include any TensorFlow native library that GPU support enabled. Since TensorFlow has removed its support of GPU
on all platforms other that Linux, we removed our platform JAR in favour of simply adding a dependency on the
`linux-x86_64-gpu` native artifact.
```xml
<dependency>
<group>org.tensorflow</group>
<artifact>tensorflow-core-native</artifact>
<version>1.0.0</version>
<classifier>linux-x86_64-gpu</classifier>
</dependency>
```
Please note that including this dependency won't work if your application also depends on `tensorflow-core-platform`. If
you need to support more platforms that Linux, you should include the other `tensorflow-core-native` dependencies
separately (see the [README](README.md) file).

### Session Run Result

In versions before 0.4.0 `Session.Runner.run` and `TensorFunction.call` returned a `List<Tensor>`. In newer versions
Expand Down

0 comments on commit 127013b

Please sign in to comment.