torch.cuda.is_available() False에서 True로 만들기

1. torch.cuda.is_available()

print(torch.cuda.is_available())

False

를 너무 많이 봐서 뇌절올꺼 같았다..ㅠㅠㅠㅠㅠㅠ

전에 NVIDIA재설치 하면서 버전을 바꾼게 문제인 것 같았다

그래서 다시한번 재설치 하기러 했다.

 

2. CUDA 재설치

https://deepflowest.tistory.com/338

[Window] CUDA 10.2 , cuDNN 7.6.5 설치

 

이 블로그를 참고해서 재설치를 진행했다.

 

3. 파이참 new project 만들기

초반에 create anaconda 가상환경 만들고

import torch

def print_hi(name):
    # Use a breakpoint in the code line below to debug your script.
    print(f'Hi, {name}')  # Press Ctrl+F8 to toggle the breakpoint.


# Press the green button in the gutter to run the script.
if __name__ == '__main__':
    print_hi('PyCharm')
    print(torch.cuda.is_available())

# See PyCharm help at https://www.jetbrains.com/help/pycharm/

터미널에

conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

해서 파이토치를 깔았다.그니까 드디어 

ㅠㅠㅠ드디어 True가 나왔다 ㅠㅠㅠㅠㅠ

눈물좔좔

4. 오늘의 교훈

버전 맞추는 것은 대단히 중요하다...