pytorch下载cuda11.1
时间: 2023-09-20 22:07:16 浏览: 223
cuda-11.1.
根据引用[1]中的信息,如果你安装的是cuda11.1,可以使用以下命令安装对应版本的pytorch:
pip3 install torch==1.8.2 cu111 torchvision==0.9.2 cu111 torchaudio===0.8.2 -f https://download.pytorch.org/whl/lts/1.8/torch_lts.html
如果以上命令报错,你可以尝试使用引用[2]中提供的其他版本的pytorch进行安装,例如:
pip install torch==1.10.1 cu111 torchvision==0.11.2 cu111 torchaudio==0.10.1 -f https://download.pytorch.org/whl/torch_stable.html
或者使用引用[3]中提供的版本进行安装:
pip install torch==2.0.0 cu118 torchvision==0.15.1 cu118 torchaudio==2.0.1 cu118 -f https://download.pytorch.org/whl/torch_stable.html
请根据你的具体需求选择合适的版本进行安装。
阅读全文