Open valentinvieriu opened 2 years ago
Yes, try changing the line
cuda: "10.2"
in cogs.yaml and then match up the lines
torch==1.9.0+cu102
torchvision==0.10.0+cu102
in requirements.txt.
Also note that cog is optional and you can instead just setup a python environment and use requirements.txt
I'm trying to run the Docker version of https://replicate.com/dribnet/pixray-api using an Nvidia 3090.
I'm getting this error
I suspect the problem is CUDA version (CUDA11.1 ) ( see https://discuss.pytorch.org/t/geforce-rtx-3090-with-cuda-capability-sm-86-is-not-compatible-with-the-current-pytorch-installation/123499/8 )
Is it possible to upgrade CUDA version ing cogs.yaml?