Guia de Instalacion de EvoX
Instalar EvoX
EvoX esta disponible en PyPI y se puede instalar mediante:
# instalar pytorch primero
# por ejemplo:
pip install torch
# luego instalar EvoX
pip install "evox[default]"
Tambien puede asignar opciones extras durante la instalacion, los extras actualmente disponibles son vis, neuroevolution, test, docs, default. Por ejemplo, para instalar EvoX con todas las caracteristicas, ejecute el siguiente comando:
pip install "evox[vis,neuroevolution]"
Instalar PyTorch con soporte de acelerador
evox depende de torch para proporcionar aceleracion por hardware.
La arquitectura general de estos paquetes Python se ve asi:
stateDiagram-v2
torch : torch
nv_gpu : NVIDIA GPU
amd_gpu : AMD GPU
cpu : CPU
direction LR
evox --> torch
torch --> nv_gpu
torch --> amd_gpu
torch --> cpu
En resumen, si evox tiene soporte para CPU o soporte para GPU Nvidia (CUDA) o soporte para GPU AMD (ROCm) depende de la version de PyTorch instalada. Consulte el sitio web oficial de PyTorch para mas ayuda con la instalacion: torch
Soporte de GPU Nvidia en Windows
EvoX soporta aceleracion por GPU a traves de PyTorch. Hay dos formas de usar PyTorch con aceleracion por GPU en Windows:
- Usando WSL 2 (Subsistema de Windows para Linux) e instalar PyTorch en el lado de Linux.
- Instalar PyTorch directamente en Windows.
Para la opcion 2, proporcionamos un script de un clic para despliegue rapido en Windows 10/11 64 bits recien instalado con GPUs Nvidia. El script no usara WSL 2 e instalara la version nativa de Pytorch en Windows. Instalara automaticamente aplicaciones relacionadas como VSCode, Git y MiniForge3.
- Asegurese de que el controlador Nvidia este correctamente instalado primero. De lo contrario, el script recurrira al modo CPU.
- Al ejecutar el script, asegure una red estable (accesible a
github.cometc.). - Si el script falla debido a un fallo de red, cierrelo y vuelva a abrirlo para continuar la instalacion.
Instalacion manual en Windows
Si prefiere instalar PyTorch directamente en Windows manualmente, puede seguir los pasos a continuacion:
- Instale el controlador Nvidia como se menciono anteriormente.
- Instale Python 3.10 o superior desde python.org.
- Instale PyTorch.
- (Opcional) Instale
triton-windowspara soporte detorch.compileen Windows. - Instale EvoX.
Windows WSL 2
Descargue el ultimo controlador de GPU NVIDIA para Windows e instalelo. Luego su WSL 2 soportara GPUs Nvidia en sus entornos Linux.
Advertencia: NO instale ningun controlador de GPU Linux de NVIDIA dentro de WSL 2. Instale el controlador en el lado de Windows.
NVIDIA tiene una [Guia de Usuario de CUDA en WSL](https://docs.nvidia.com/cuda/wsl-user-guide/index.html) detallada
Soporte de GPU AMD (ROCm)
Recomendamos usar un contenedor Docker de rocm/pytorch.
docker run -it --network=host --device=/dev/kfd --device=/dev/dri --group-add=video --ipc=host --cap-add=SYS_PTRACE --security-opt seccomp=unconfined --shm-size 8G -v $HOME/dockerx:/dockerx -w /dockerx rocm/pytorch:latest
Verificar la instalacion
Abra una terminal de Python y ejecute lo siguiente:
from torch.utils.collect_env import get_pretty_env_info
import evox
print(get_pretty_env_info())