Skip to content

Instantly share code, notes, and snippets.

@tommylees112
Last active June 28, 2021 13:15
Show Gist options
  • Select an option

  • Save tommylees112/1617759a2fd8e886d07bd05d6c0e62fb to your computer and use it in GitHub Desktop.

Select an option

Save tommylees112/1617759a2fd8e886d07bd05d6c0e62fb to your computer and use it in GitHub Desktop.
Create working environment on virtual machine
conda create -n ml python=3.8
source activate ml
conda install pytorch torchvision torchaudio cudatoolkit=11.2 -c pytorch -c nvidia --yes
conda install -c conda-forge seaborn=0.11 --yes
conda install -c conda-forge netcdf4 numba tqdm jupyterlab tensorboard ipython pip ruamel.yaml xarray descartes statsmodels scikit-learn black mypy --yes
pip install geopandas
ipython --pdb -c "import torch; assert torch.cuda.is_available(); print(torch.backends.cudnn.version()); print(torch.cuda.device_count());"
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment