Python and Pytorch Version for Train #250
Reference in New Issue
Block a user
Delete Branch "%!s()"
Deleting a branch is permanent. Although the deleted branch may continue to exist for a short time before it actually gets removed, it CANNOT be undone in most cases. Continue?
i have newbie question, before update on train model im using insightface 0.2.1 python 3.9 and january 2022 pytorch. its still work for swap video (no error) and image (error word but still have good result). then pytorch update on lastest, got all of error on them. iam using aws cloud gpu maybe its default to python 3.9.
thanks
If unsure about how to install, always run using the preparation documentation
Using Anaconda allows you to install whatever version of Python you want in a "container" environment. That way, you don't have to mess around with AWS version of Python.
Also, if you have errors with Pytorch, it's most likely that version is not compatible with the GPU you're using. You have to search for a compatible version here.
For example, the newer GPUs like the 3000 series don't support CUDA 10.1, so you have to install 11.1 when installing torch. Make sure your Pytorch version matches the one in the preparation documentation.
The main part of new available training scripts have no relationship with insightface respo.
Only the face detection function depend on insightface.
Face detection is only performed when processing the unaligned picture.
A few tips:
-Try not to mix and match your cuda versions.
-Make sure you're installing pytorch with cuda, from the pytorch official website.
-You might even want to set up a different environment for training Incase you're setting up tensorRt.
Also, do this in terminal:
python
Import torch
torch.cuda.is_available()
-This should return true. You might be having compatibility issues with python 3.9; in which case just create a new environment with python 3.6 or 3.7
Good luck.