Naver-AI-Hackathon / cs492I

2 stars 0 forks source link

Fail to building docker image #47

Open jyeongho opened 3 years ago

jyeongho commented 3 years ago

Like issue in #14 , I've got segmentation fault error. So, I tried to change docker environment from "nsml/ml:cuda10.1-cudnn7-pytorch1.3keras2.3" to "nvcr.io/nvidia/pytorch:20.03-py3". image

However, when I run a session, I got this error message. image

How can I change docker environment? Should I enter the docker image name into docker image cache sheet in "[https://docs.google.com/spreadsheets/d/1OHsW3HUhL6dSZ5KaZSICpW3MpEeh0Sykrqd883CtQ8Q/edit#gid=1078177844]"?)

myoons commented 3 years ago

@jyeongho How did you solve this problem?

jyeongho commented 3 years ago

@jyeongho How did you solve this problem?

image

I solved this problem by adding 'torch==1.7.0' and 'torchvision==0.8.1' in setup.py file.

myoons commented 3 years ago

@jyeongho Thanks!