Spaces:
Sleeping
Sleeping
| with import <nixpkgs> {}; | |
| mkShell { | |
| name = "tensorflow-cuda-shell"; | |
| buildInputs = with python3.pkgs; [ | |
| pip | |
| setuptools | |
| python39Packages.tensorflow | |
| python39Packages.numpy | |
| ]; | |
| shellHook = '' | |
| export LD_LIBRARY_PATH=${pkgs.stdenv.cc.cc.lib}/lib:$LD_LIBRARY_PATH | |
| alias pip="PIP_PREFIX='$(pwd)/_build/pip_packages' TMPDIR='$HOME' \pip" | |
| export PYTHONPATH="$(pwd)/_build/pip_packages/lib/python3.7/site-packages:$PYTHONPATH" | |
| export PATH="$(pwd)/_build/pip_packages/bin:$PATH" | |
| unset SOURCE_DATE_EPOCH | |
| ''; | |
| } | |