[딥러닝/파이토치] m1 pytorch gpu

김진만·2023년 6월 25일
0

AI

목록 보기
2/2

m1 , pytorch GPU 활성화하기

m1, m2 : arm64 아키텍처 사용자라면
텐서플로우와 파이토치로 로컬에서 GPU를 사용하고 싶고
사용할 수 있다!!
하지만 어떻게~?!?!
tensorflow에 관한 게시글은 너무 많아서 패스하고
토치에 대해 간단하게 설명을 하면
일단 Tensorflow 같은 경우는 conda설치( Miniforge) 후에 가상환경 만들고 관리하고 환경변수 설정도 해줬던 것 같음 ( bashrc, export ) 등등
pytorch도 비슷하게 conda에 가상환경 만들었는데 안되는거임 버전이슈인가 했는데.
그래서 그냥 conda 가상환경에서 나오고

깨알 conda 명령어
가상환경 종류 확인
conda info --envs
가상환경 생성
conda create --name <이름> <옵션줄수있음,파이썬버전등,나는잘안줌)
가상환경 나오기
conda deactivate

로컬에 가상환경 python -m venv <가상환경이름>
해서 거기서

pip install torch
python

쉘 드가자!!

import torch
torch.backends.mps.is_avaiable()


True 뜨면 끝!
GPU 활성상태 확인은
is_available() 구절을 참 좋아함 ㅋ

profile
충분한 전기와 컴퓨터 한 대와 내 이 몸만 남아 있다면 지구를 재건할 수 있습니다.

0개의 댓글