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())
'빅데이터' 카테고리의 다른 글
| Polars (0) | 2023.11.21 |
|---|---|
| 패키지 설치 시 코드 에러 문제 (0) | 2023.10.15 |
| MapReduce와 쿼리튜닝 (0) | 2023.09.27 |
| pgRouting 함수 (0) | 2023.09.27 |
| QGIS 수치표고모델 (0) | 2023.09.27 |