테스트된 환경(Tested Environments)
- Ubuntu 20.04
- Cuda 11.2
- GPU: Rtx3090 x2
GPU 버전 실행시 반드시 NVIDIA Docker도 같이 설치(Must pull nvidia/cuda docker in order to work properly)
목록 링크: nvidia/cuda
위의 링크에서 GPU 버전에 맞는 CUDA VERSION을 확인 후 cuda runtime docker를 pull 합니다.
e.g) ubuntu 20.04 버전의 CUDA 11.2 버전의 경우
docker pull nvidia/cuda:11.2.1-runtime-ubuntu20.04
주요 패키지(Pre-installed packages)
- gensim 3.8.3
- kaggle 1.5.10
- Keras 2.4.3
- Keras-Preprocessing 1.1.2
- konlpy 0.5.2
- lightgbm 3.1.1
- matplotlib 3.3.4
- mecab-python 0.996-ko-0.9.2
- mlflow 1.14.0
- mlxtend 0.18.0
- nltk 3.5
- notebook 6.2.0
- numpy 1.19.5
- pandas 1.1.5
- pip 21.0.1
- pycaret 2.3.0
- scikit-learn 0.23.2
- scipy 1.5.4
- seaborn 0.11.1
- surprise 0.1
- tensorboard 2.4.1
- tensorflow 2.4.0
- tensorflow-datasets 4.2.0
- tensorflow-gpu 2.4.1
- torch 1.7.1+cu110
- torchaudio 0.7.2
- torchvision 0.8.2+cu110
- wordcloud 1.8.1
- xgboost 1.4.0-SNAPSHOT