1. torch.cuda.is_available()
print(torch.cuda.is_available())
False
๋ฅผ ๋๋ฌด ๋ง์ด ๋ด์ ๋์ ์ฌ๊บผ ๊ฐ์๋ค..ใ ใ ใ ใ ใ ใ
์ ์ NVIDIA์ฌ์ค์น ํ๋ฉด์ ๋ฒ์ ์ ๋ฐ๊พผ๊ฒ ๋ฌธ์ ์ธ ๊ฒ ๊ฐ์๋ค
๊ทธ๋์ ๋ค์ํ๋ฒ ์ฌ์ค์น ํ๊ธฐ๋ฌ ํ๋ค.
2. CUDA ์ฌ์ค์น
https://deepflowest.tistory.com/338
์ด ๋ธ๋ก๊ทธ๋ฅผ ์ฐธ๊ณ ํด์ ์ฌ์ค์น๋ฅผ ์งํํ๋ค.
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. ์ค๋์ ๊ตํ
๋ฒ์ ๋ง์ถ๋ ๊ฒ์ ๋๋จํ ์ค์ํ๋ค...
'KAIST MASTER๐ > python' ์นดํ ๊ณ ๋ฆฌ์ ๋ค๋ฅธ ๊ธ
ํ์ด์ฌ % - ๋ฌธ์์ด % ํฌ๋งคํ (0) | 2021.01.22 |
---|---|
tqdm ์ด๋ - ํ์ด์ฌ ์ํ ๋ฐ ๋ผ์ด๋ธ๋ฌ๋ฆฌ (0) | 2021.01.22 |
๋๊ธ