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

2
.python-version
View File

@ -1 +1 @@
3.10.5
3.10.5

7315
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://discuss.tensorflow.org/t/tensorflow-io-gcs-filesystem-with-windows/18849/4
# 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 = "^2.14.0"
tensorflow = {version = "^2.14.0", extras = ["and-cuda"]}
deepface = "^0.0.79"