Tensorflow: Difference between revisions
imported>Mjlbach mNo edit summary |
Sv-cheats-1 (talk | contribs) Updated wiki page on tensorflow. Old configuration didn't work anymore, so I provided one that worked for me |
||
(9 intermediate revisions by 3 users not shown) | |||
Line 1: | Line 1: | ||
= Tensorflow = | |||
=== Quick shell <ref name=":0">https://discourse.nixos.org/t/cuda-tensorflow-my-setup-is-really-hacky-would-appreciate-help-unhackying-it/43912/2</ref> === | |||
nix-shell --arg config '{ cudaSupport = true; allowUnfree = true; }' -p 'python3.withPackages (ps: [ ps.tensorflow ]<code>)'</code> | |||
This starts a shell with tensorflow and it has GPU support. | |||
=== Shell config === | |||
A basic shell config would look something like this: | |||
<code># shell.nix | |||
{ pkgs ? import <nixpkgs> {}, config ? {} }: | |||
let | |||
<nowiki> </nowiki> pythonPackages = pkgs.python3.withPackages (ps: [ ps.tensorflow ]); | |||
in | |||
pkgs.mkShell { | |||
<nowiki> </nowiki> name = "tf"; | |||
<nowiki> </nowiki> buildInputs = [ | |||
<nowiki> </nowiki> (pythonPackages) | |||
<nowiki> </nowiki> ]; | |||
<nowiki> </nowiki> shellHook = <nowiki>''</nowiki> | |||
<nowiki> </nowiki> export PYTHONPATH="${pythonPackages}:${PYTHONPATH:-}" | |||
<nowiki> </nowiki> <nowiki>''</nowiki>; | |||
}</code> | |||
To make this configuration work, edit <code>~/.config/nxpkgs/config.nix</code>: | |||
<code># ~/.config/nxpkgs/config.nix | |||
{ cudaSupport = true; allowUnfree = true; }</code> | |||
== References == | |||
<ref name=":0" /> | |||