박가방

cuda, pythorch 설치 정리 본문

프로그램 언어/Python

cuda, pythorch 설치 정리

박가방 2023. 7. 10. 21:56

1.스펙

rtx 2070 notebook

 

2. cuda_11.2 설치

 

3. graphic driver 설치

 

4.1 cudnn-11.2-windows-x64-v8.1.0.77 설치후

 

bin lib include -> cuda 폴더로 복사 붙여넣기

 

4.2 환경 변수 경로 설정

5. 가상환경 만들기

pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 -f https://download.pytorch.org/whl/torch_stable.html

6. cuda 버전에 맞는 torch 설치

 

7. 가상환경 만들기

python -m ipykernel install --user --name 가상환경이름 --display-name 가상환경이름

 

8. Jupyter lab 실행

가상환경 커널 연결후

 

확인

 

9.  잘돌아간다 ㅜㅜ

 

 

 

참고 :

https://deepflowest.tistory.com/338 

 

[Window] CUDA 10.2 , cuDNN 7.6.5 설치

■ 기존 버전, 환경 확인 OS : Windows 10 Graphic card : GTX1080 CUDA 11.1 ■ 설치 리스트 CUDA 10.2 cuDNN 7.6.5 1. 기존 CUDA, NVIDIA 드라이버 삭제 1) "프로그램 추가/제거" 에서 "NVIDIA" 검색 후 관련된 것들 모두 삭

deepflowest.tistory.com

 

https://github.com/moaaztaha/Yolo-Interface-using-Streamlit

'프로그램 언어 > Python' 카테고리의 다른 글

no module named speech_recognition  (0) 2023.07.10
[python] jupyter lab attr 오류  (0) 2023.07.10
[python] requirements.txt 설치 오류  (0) 2023.07.10
[python] tensorflow-gpu, ssl 오류  (0) 2023.07.10
[오류] tensorflow 설치 오류  (0) 2023.04.13
'프로그램 언어/Python' Related Articles more