envd icon indicating copy to clipboard operation
envd copied to clipboard

bug: torch installed without cuda support

Open aseaday opened this issue 2 years ago • 7 comments

Description

def build():
    base(os="ubuntu20.04", language="python3.8")
    install.cuda(version="11.2.2", cudnn="8")
    install.conda_packages(
        name=[
            "pytorch==1.12.1",
            "cudatoolkit=10.2",
        ],
        channel=["pytorch"],
    )

https://pytorch.org/get-started/previous-versions/

aseaday avatar Nov 30 '22 05:11 aseaday

/cc @VoVAllen @kemingy

gaocegege avatar Nov 30 '22 05:11 gaocegege

image

aseaday avatar Nov 30 '22 05:11 aseaday

What's the error message?

VoVAllen avatar Nov 30 '22 06:11 VoVAllen

I failed to reproduce failed with initial frozen solve image image

cutecutecat avatar Nov 30 '22 07:11 cutecutecat

It may be caused by cache. The error is that I didn't install a torch with cuda support.

aseaday avatar Nov 30 '22 07:11 aseaday

Why do you need install.cuda(version="11.2.2", cudnn="8") and "cudatoolkit=10.2" at the same time?

kemingy avatar Nov 30 '22 07:11 kemingy

I don't know wether we can install.cuda version 10.2

aseaday avatar Nov 30 '22 07:11 aseaday