slashtechno
/
wyzely-detect
mirror of https://github.com/slashtechno/wyzely-detect
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Somehow got CUDA working

This commit is contained in:
slashtechno 2023-12-08 16:26:04 -06:00
parent ecf47a05aa
commit fc943644fc
Signed by: slashtechno
GPG Key ID: 8EC1D9D9286C2B17
3 changed files with 3692 additions and 3632 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2085
poetry.lock generated
View File

File diff suppressed because it is too large Load Diff

7
pyproject.toml
View File

@ -27,8 +27,13 @@ torch = ">=2.0.0, !=2.0.1, !=2.1.0"
# https://stackoverflow.com/a/76477590/18270659 # https://stackoverflow.com/a/76477590/18270659
# https://discuss.tensorflow.org/t/tensorflow-io-gcs-filesystem-with-windows/18849/4 # https://discuss.tensorflow.org/t/tensorflow-io-gcs-filesystem-with-windows/18849/4
# Might be able to remove this version constraint later # Might be able to remove this version constraint later
# Working versions:
# Python version 3.10.12 and 3.10.5 both work
# CUDA version - 12.2
# cuDNN version - 8.8.1
# Installed from Nvidia website - nvidia-cuda-toolkit is not installed, but default PopOS drivers are installed
tensorflow-io-gcs-filesystem = "0.31.0" tensorflow-io-gcs-filesystem = "0.31.0"
tensorflow = "^2.14.0" tensorflow = {version = "^2.14.0", extras = ["and-cuda"]}
deepface = "^0.0.79" deepface = "^0.0.79"