envd
envd copied to clipboard
bug: torch installed without cuda support
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/
/cc @VoVAllen @kemingy

What's the error message?
I failed to reproduce failed with initial frozen solve

It may be caused by cache. The error is that I didn't install a torch with cuda support.
Why do you need install.cuda(version="11.2.2", cudnn="8") and "cudatoolkit=10.2" at the same time?
I don't know wether we can install.cuda version 10.2