cuda 셋팅

pipenv에서의 pipfile 셋팅

[[source]]
url = "https://pypi.org/simple"
verify_ssl = true
name = "pypi"

[[source]]
url = "https://download.pytorch.org/whl/cu118"
verify_ssl = false
name = "pytorch"

[packages]
keras-segmentation = "*"
tensorflow = "*"
torch = {version = "*", index = "pytorch"}
tensorflow=""

[dev-packages]

[requires]
python_version = "3.10"

 

tensorflow에서 확인하는 방법

import tensorflow as tf
print("TensorFlow 버전:", tf.__version__)
print("CUDA 지원 여부:", tf.test.is_built_with_cuda())
print("GPU 사용 가능 여부:", tf.test.is_gpu_available(cuda_only=False, min_cuda_compute_capability=None))
print("CUDA 버전 정보:", tf.test.gpu_device_name())