[Google colab] 개인이 생성한 모듈을 colab에서 import해서 사용하고자 하는 경우 / (+ 데이터 업로드 하기)

    컴퓨터 자원 문제나, 파이토치나 텐서플로우의 버전 맞추는 번거로움 때문에, colab을 통해서 코드를 돌려보는 경우가 많다. 개인이 만든 모듈을 Colab에서 import해서 사용하고자 하는 경우는 다음의 코드를 사용하면 된다.

# 아래의 코드를 돌리면, 파일 선택이라는 버튼이 생성되며 로컬 컴퓨터에서 사전에 만든 모듈을 업로드 하면 된다. 

from google.colab import files
src = list(files.upload().values())[0]

# 나의 경우, NGCF 모듈을 업로드 했으며,
#import NGCF 했을 때 모듈이 찾아지지 않는다는 에러가 뜨지 않으면 성공한 것이다. 

 

    한편으로, 위의 코드는 Colab에서 사용할 데이터를 업로드 하는데도 사용할 수 있다. 하지만, 이 방식은 런타임을 재시작할 때마다 업로드한 파일들은 소실되므로 그리 추천할만한 방법은 아니다. google.colab의 drive의 모듈을 통해서 자신의 구글 드라이브에 있는 파일들을 손쉽게 연결하여 사용할 수 있다. 

과정은 다음과 같다. 

1. google drive에 파일을 업로드 한다.

 

2. 다음의 코드를 실행한다.

from google.colab import drive
drive.mount('/content/gdrive')

 

3. 위의 코드를 실행하면, 연결된 링크에는 개인의 계정의 구글 드라이브 파일을 사용할 수 있는 권한 코드 정보가 있다. 

'*********'(조금 긴 번호)를  아웃풋 쉘에 입력하면, 연결 성공.

4. '/content/gdrive/My Drive/'의 경로를 사용하여 자신의 파일에 접근하면 된다.

5. 나의 경우, 내가 작업한 패키지를 python 명령어로 .py를 실행하기 위해서 작업 경로를 패키지 경로로 바꾸었다. 

%cd /content/gdrive/My Drive/내가 만든 패키지 이름/

 

덧) 아래의 코드로 드라이브의 패키지 파일과 데이터를 임시 작업 폴더 colab에 옮기게 되면, colab의 런타임이 꺼지게 되면 구글 드라이브에 업로드한 파일도 삭제된다. 

* shutil.move ('Mydrive/내가 만든 패키지 이름/, '/content/')