adeytown75/pytorch repository overview

CUDA 9.0 enabled versions of PyTorch with various additional packages are available

0.41-cude6-cudnn7-deve-sckitimage

• This version includes the additional python3 packages: matplotlib, numpy, sklearn, scikit-image and comet_ml

0.4.1-cuda9-cudnn7-devel-astro

• This version includes the additional python3 packages: astropy, h5py, numpy, scipy, pandas, snakeviz and statsmodels