Skip to content

Instantly share code, notes, and snippets.

@konradish
Last active October 2, 2021 17:14
Show Gist options
  • Save konradish/6663bd6f287b9791f2b6905f3ffa0d7e to your computer and use it in GitHub Desktop.
Save konradish/6663bd6f287b9791f2b6905f3ffa0d7e to your computer and use it in GitHub Desktop.
Set up tensorflow GPU on Windows
REM For new versions check out the grid at https://www.tensorflow.org/install/source_windows#gpu
conda create -n tf python=3.8.7
REM Or to upgrade: conda install -c anaconda python=3.8
conda install -y -c anaconda tensorflow-gpu==2.5.0
conda install -y jupyter seaborn matplotlib numpy keras-gpu pandas tqdm tensorflow_addons
conda install -y -c conda-forge ipywidgets
conda install -y jupyterlab
jupyter labextension install @jupyter-widgets/jupyterlab-manager
conda install -y -c conda-forge matplotlib
conda install -y -c conda-forge keras
conda install -y opencv
conda install -y tqdm
conda install -y h5py==3.4.0
conda install -y scikit-image
conda install -y tensorflowjs
@konradish
Copy link
Author

I think you need a Visual Studio C compiler, but you probably DON'T need to download cuDNN or any of the other stuff from NVidia's website like all the online guides tell you. That gets installed as a dependency with tensorflow-gpu

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment