记录安装pytorch遇见的坑,主要是解决torch.cuda.is_available()返回False的问题。
문제는 다음과 같습니다
. 이 문제는 너무 형이상학적입니다. 다양한 솔루션을 시도했지만 여전히 해결되지 않습니다. 여기에 기록해 두겠습니다. 인터넷에 있는 방법은 기본적으로 conda를 사용하여 다운로드됩니다. 여기 테스트는 모두 실패했기 때문에 pip는 다운로드에 사용되며 다운로드 후 테스트에 성공했습니다.
1. 컴퓨터 버전
GPU: GTX050Ti
CUDA 버전: 12.0(이전에는 11.6이었는데 왜 자동으로 12.0으로 업그레이드되는지 모르겠습니다. 이 명령을 사용하여 cuda를 다운로드할 수도 있습니다.)
드라이버 버전: 526.47
2. 인터넷 문의
2.1 가능한 이유
1. CPU 버전을 다운로드했습니다.
2. 버전이 잘못되었습니다.
3. 솔루션
3.1 방법 1 pip를 사용하여 설치
conda가 자동으로 cpu 버전을 다운받은 것일 수도 있으니 pip로 설치
pip install torch torchvision torchaudio --pre --extra-index-url https://download.pytorch.org/whl/nightly/cu116
pytorch_pip라는 환경을 만들었습니다. 이 코드를 사용하여
다운로드 후 gpu를 사용할 수 있는지 확인하면 성공했습니다! ! ! ! ! ! ! ! !