torch.cuda.is_available()이 False를 반환하는 문제 해결(pip를 사용하여 성공적으로 설치)

记录安装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를 사용할 수 있는지 확인하면 성공했습니다! ! ! ! ! ! ! ! !

여기에 이미지 설명 삽입

추천

출처blog.csdn.net/qq_43471945/article/details/127703461