conda install pytorch torchvision gpuonly
时间: 2023-12-04 10:04:33 浏览: 128
Linux系统安装pytorch和torchvision:torch1.8.0+torchvision0.9.0
这个命令是用来在 conda 环境中安装 PyTorch 和 TorchVision,并且只会安装 GPU 版本。其中,PyTorch 是一个专门针对深度学习任务的开源机器学习框架,而 TorchVision 则是 PyTorch 的一个扩展库,提供了一些用于计算机视觉任务的常用工具和数据集。
需要注意的是,此命令需要在支持 GPU 计算的机器上运行,且需要安装 CUDA 和 cuDNN 等 GPU 相关的软件包。如果没有 GPU,可以使用 `conda install pytorch torchvision` 命令安装 CPU 版本。
阅读全文