conda install pytorch torchvision torchaudio cudatoolkit=10.1 -c pytorch
时间: 2023-09-16 16:10:54 浏览: 333
torch和torchvision
5星 · 资源好评率100%
这是安装 PyTorch 和相关的工具包的命令,其中包括 torchvision 和 torchaudio。这个命令还指定了要安装的 CUDA 工具包的版本为 10.1,这个版本需要在你的计算机上安装了对应版本的 CUDA 驱动才能正常使用。该命令使用了 PyTorch 官方提供的 conda 渠道进行安装。
阅读全文