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