Skip to content

Instantly share code, notes, and snippets.

@tabrez
Last active August 11, 2023 09:44
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save tabrez/3f9605fafc5d4cedb9536764347bdbe7 to your computer and use it in GitHub Desktop.
Save tabrez/3f9605fafc5d4cedb9536764347bdbe7 to your computer and use it in GitHub Desktop.
Setup a mamba env with fastai, pytorch, cuda, jupyterlab libraries
#!/bin/bash
# tested with mamba version 1.4.2
# wget "https://github.com/conda-forge/miniforge/releases/latest/download/Mambaforge-$(uname)-$(uname -m).sh"
# bash Mambaforge-$(uname)-$(uname -m).sh -b
mamba init
mamba create -n fastbook -y python=3.11.4
source activate fastbook
mamba install --yes -c conda-forge ipywidgets=8.1.0 sentencepiece=0.1.99 nbdev=2.2.10 \
jupyterlab=4.0.4 jupyterlab-autosave-on-focus-change=0.2.0
mamba install --yes -c fastchan -c pytorch -c nvidia \
fastbook=0.0.28 pytorch=2.0.1 torchvision=0.15.2 torchaudio=2.0.2 pytorch-cuda=11.7
jupyter labextension disable "@jupyterlab/apputils-extension:announcements"
md "{{ home }}/mambaforge/envs/fastbook/share/jupyter/lab/settings"
cp overrides.json "{{ home }}/mambaforge/envs/fastbook/share/jupyter/lab/settings/overrides.json"
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment