Pytorch 가 지정 한 GPU

771 단어 pytorchdeeplearning
전환 하 다 http://www.cnblogs.com/darkknightzh/p/6836568.html
 
PyTorch 는 기본적으로 0 부터 시작 하 는 GPU 를 사용 하 며, GPU 0 이 프로그램 을 실행 중인 경우 다른 GPU 를 지정 해 야 합 니 다.
사용 할 GPU 를 지정 하 는 방법 은 다음 과 같다.
1. tensorflow 가 지정 한 GPU 와 같은 방식 으로 CUDA 사용VISIBLE_DEVICES。
1.1 직접 터미널 에서 설정:
CUDA_VISIBLE_DEVICES=1 python my_script.py

1.2 python 코드 에서 설정:
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "2"

사이트 주소 참조:http://www.cnblogs.com/darkknightzh/p/6591923.html
2. 함수 사용 set_device
import torch
torch.cuda.set_device(id)

이 함 수 는 pytorch - master \ \ torch \ cuda \ \ 를 보십시오.init__.py。
하지만 공식 적 으로 는 CUDA 사용 을 권장 합 니 다.VISIBLE_DEVICES, 사용 권장 하지 않 음 set_device 함수

좋은 웹페이지 즐겨찾기