gpu_server_setup/setup_tensorflow_conda.sh

12 lines
315 B
Bash

#!/bin/bash
set -e
if [ "$EUID" -eq 0 ]
then echo "Please run not as root"
exit
fi
conda create ipykernel --name tensorflow -c conda-forge
conda install --yes --name tensorflow tensorflow-gpu tensorboard -c conda-forge
python -m ipykernel install --user --name tensorflow --display-name "Python (Tensorflow)"