Pytorch 가 지정 한 GPU
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 함수
이 내용에 흥미가 있습니까?
현재 기사가 여러분의 문제를 해결하지 못하는 경우 AI 엔진은 머신러닝 분석(스마트 모델이 방금 만들어져 부정확한 경우가 있을 수 있음)을 통해 가장 유사한 기사를 추천합니다:
정확도에서 스케일링의 영향데이터셋 스케일링은 데이터 전처리의 주요 단계 중 하나이며, 데이터 변수의 범위를 줄이기 위해 수행됩니다. 이미지와 관련하여 가능한 최소-최대 값 범위는 항상 0-255이며, 이는 255가 최대값임을 의미합니다. 따...
텍스트를 자유롭게 공유하거나 복사할 수 있습니다.하지만 이 문서의 URL은 참조 URL로 남겨 두십시오.
CC BY-SA 2.5, CC BY-SA 3.0 및 CC BY-SA 4.0에 따라 라이센스가 부여됩니다.