mirror of
https://github.com/KoboldAI/KoboldAI-Client.git
synced 2024-12-12 00:35:41 +01:00
03501a4c8c
Looks like cudatoolkit is now shipping 11.3, but pytorch has no version for this resulting in an installation of the CPU only version. This is going to lead to people unable to get their GPU running, so for now we force the recommended 11.1 version. I also don't see any harm in allowing Python 3.9 so thats now the default as well to prevent future issues.
17 lines
254 B
YAML
17 lines
254 B
YAML
name: koboldai
|
|
channels:
|
|
- huggingface
|
|
- pytorch
|
|
- conda-forge
|
|
- defaults
|
|
dependencies:
|
|
- colorama
|
|
- flask-socketio
|
|
- pytorch
|
|
- python=3.9.*
|
|
- cudatoolkit=11.1
|
|
- tensorflow-gpu
|
|
- transformers
|
|
- pip
|
|
- pip:
|
|
- flask-cloudflared |