mistralai / mistral-inference

Official inference library for Mistral models
https://mistral.ai/
Apache License 2.0
9.49k stars 835 forks source link

Fix Dockerfile #81

Open nicholasjpaterno opened 9 months ago

nicholasjpaterno commented 9 months ago

Latest PyTorch version is built with CUDA 12.1

Pin the version to the latest stable release built with CUDA 11.8