Open YeongJunKim opened 4 years ago
import tensorflow as tf
print(tf.__version__)
# 1.14.0
tf.test.is_gpu_available(
cuda_only=False,
min_cuda_compute_capability=None
)
# True
from keras import backend as K
print(keras.__version__)
# 2.2.4
K.tensorflow_backend._get_available_gpus()
# ['/job:localhost/replica:0/task:0/device:GPU:0']
import torch
torch.cuda.device_count()
# 1
torch.cuda.get_device_name(0)
# GeForce RTX 2080 Ti
torch.cuda.is_available()
# True
기본가이드
텐서플로우 홈페이지 gpu 설치
CUDA downgrade
텐서플로우 gpu 사용 여부 확인
conda PermissionError 13
Blacklist the driver
Confirm the content of the new modprobe config file
Update kernel initramfs
Reboot
Remove Any Other NVIDIA Driver
Add the graphics driver PPA
Purge the driver
Reboot
Add Nvidia package repositories
Install Nvidia driver
Reboot
Install Runtime & Development Libraries (cuDNN)
Install TensorRT
check nvidia
Install Anaconda package
click here
If conda has permission error 13
conda python3
vertural env
tensorflow & keras install