AI-GrandChallenge / round-1

14 stars 11 forks source link

setup.py 설정 문의드립니다. #32

Open KOR-PAIR opened 4 years ago

KOR-PAIR commented 4 years ago

아래 사용하는 패키지이며, CUDA는 10.1, Tensorflow-gpu는 1.9 사용합니다. import operator import tensorflow as tf import numpy as np import os import scipy.io import glob import scipy.misc as misc from imageio import imread from PIL import Image import time import shutil import random from datetime import datetime, timedelta import pandas as pd import skimage import skimage.io import skimage.transform

NeuroAI-lab commented 4 years ago

tensorflow 1.9 버전이 아닌 다른버전을 사용하려 할때는 다음과 같이 변경을 하여 사용하면 되는 건가요?

nsml: tensorflow/tensorflow:1.15.0-gpu-py3

nsml-admin commented 4 years ago

@kw-nisp 네 setup.py에 말씀하신대로 적어주시면 됩니다 (tensorflow/tensorflow:1.15.0-gpu-py3 이미지가 docker hub에 있다는 가정하에)

#nsml: tensorflow/tensorflow:1.15.0-gpu-py3
from distutils.core import setup

setup(
    name='iitp_trash',
    version='1.0',
    install_requires=[
                      'tqdm',
                      'pillow',
                      ]
)