machine learning

GeForce MT150 사용자 pytorch 설치

민동기 2020. 6. 7. 21:50
728x90

1. pytorch 설치 

greeksharifa.github.io/pytorch/2018/11/02/pytorch-usage-01-introduction/

2. MT150 (보통 노트북에 장착된 GPU) 에 맞는 CUDA 설치

github.com/pytorch/pytorch/issues/28321

 

CUDA unavailable when pytorch 1.3.0. installed with cudatoolkit 10.1 · Issue #28321 · pytorch/pytorch

🐛 Bug torch.cuda.is_available returns false To Reproduce simply install pytorch conda install pytorch torchvision cudatoolkit=10.1 -c pytorch Expected behavior cuda being available Environment Coll...

github.com