• Joseph Young's avatar
    Update PyTorch wheel source to CUDA 11.8 · 2588da0e
    Joseph Young authored
    Modified the Dockerfile to install PyTorch, torchvision, and torchaudio from a CUDA 11.8 specific wheel URL. This ensures compatibility with the CUDA version in our environment and potentially improves performance and stability for GPU-accelerated operations.
    2588da0e
Dockerfile-cuda 3.88 KB