본문 바로가기

Shakerato

검색하기
Shakerato
프로필사진 Shakeratto

  • 분류 전체보기 (77)
    • Research (67)
    • Photography (1)
Guestbook
Notice
«   2025/05   »
일 월 화 수 목 금 토
1 2 3
4 5 6 7 8 9 10
11 12 13 14 15 16 17
18 19 20 21 22 23 24
25 26 27 28 29 30 31
Tags
  • Deep Learning
  • colaboratory
  • colab
  • pyTorch
  • python
  • YouTube 8M
  • windows
  • Anaconda
  • dataset
  • urllib
  • dlib
  • ppc64le
  • download
  • keras
  • 딥러닝
  • ubuntu
  • face_recognition
  • object detection
  • Windows 10
  • FIle
  • Jupyter notebook
  • python3
  • gpu memory
  • raspberry pi
  • shakeratos
  • linux
  • TensorFlow
  • error
  • CUDA
  • install
more
Archives
Today
Total
관리 메뉴
  • 글쓰기
  • 방명록
  • RSS
  • 관리

목록Memory (1)

Shakerato

GPU devices checking tips for Deep Learning (Be careful!)

[Tensorflow]from tensorflow.python.client import device_libprint(device_lib.list_local_devices()) [Keras]from keras import backend as KK.tensorflow_backend._get_available_gpus() [Pytorch]import torchtorch.cuda.get_device_name(0) # number of gpu (cuda:0)torch.cuda.get_device_name(1) # number of gpu (cuda:1) Be careful with the above codes! The code for checking device(GPU) takes your GPU memory a..

Research 2019. 1. 16. 17:44
Prev 1 Next

전체 방문자수 : 오늘 방문자수 : 어제 방문자수 :

티스토리툴바