Tensorflow: Difference between revisions
imported>Mjlbach Created page with "== Tensorflow == There some possible ways to install tensorflow. Nixpkgs provides multiiple versions, however, it is often desirable to be able to install the latest nightly..." |
imported>Mjlbach mNo edit summary |
||
| Line 5: | Line 5: | ||
* By making a nix-shell | * By making a nix-shell | ||
<syntaxhighlight lang="nix"> | |||
with import <nixpkgs> {}; | with import <nixpkgs> {}; | ||
mkShell { | mkShell { | ||
| Line 22: | Line 22: | ||
''; | ''; | ||
} | } | ||
</ | </syntaxhighlight> | ||
Within this shell, pip install tf-nightly should work and provide GPU support. The cuda toolkit version can be changed to correspond with the matching tensorflow version. | Within this shell, pip install tf-nightly should work and provide GPU support. The cuda toolkit version can be changed to correspond with the matching tensorflow version. | ||