tensorflow설정1 [tensorflow 설정] gpu 세팅하기 pytorch로 이미지 분석을 하려다 보니 생각보다 자료가 많이 없더라구요. 그래서 먼저 tensorflow로 모델을 제작하고 그후에 pytorch로 진행을 하고자 합니다. pytorch는 생각보다 쉽게 세팅이 가능하더라구요. 학교에서는 GTX1050ti, 집에서는 GTX1060 으로 진행하였습니다. 세팅의 경우 똑같습니다. 위 두개의 GPU에서 CUDA는 11.0을 이용하였고, cuDNN의 경우 8.0.5를 이용하였습니다. 그리고 pip install을 진행할 때 다음과 같이 진행해줍니다. pip install tensorflow==2.4.0 pip install tensorflow-gpu==2.4.0 그리고 나서 python을 틀고 tensorflow를 import시키면 성공했다는 무엇인가가 나오고 .. 2022. 3. 10. 이전 1 다음
