cuda11.2 conda对应pytorch指令
时间: 2023-06-08 07:07:54 浏览: 163
浅谈pytorch、cuda、python的版本对齐问题
5星 · 资源好评率100%
您好,针对您的问题,可以使用以下conda指令安装适用于cuda 11.2的pytorch:
conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c nvidia
请注意,这里使用了11.1版本的cudatoolkit,这是因为cuda 11.2还没有被pytorch官方支持。如果您需要使用最新的cuda 11.2,可以考虑使用源码编译pytorch。
希望对您有所帮助。
阅读全文