您可以使用以下命令来在conda中安装PyTorch:
conda install pytorch torchvision torchaudio -c pytorch
该命令将安装PyTorch、TorchVision和TorchAudio包。
需要注意的是,您需要选择适合您的操作系统和CUDA版本的PyTorch安装包。因此,在上面的命令中,您可能需要更改“cudaXX”部分,其中XX是您要安装的CUDA版本。
例如,如果您使用CUDA 11.0,则命令如下:
conda install pytorch torchvision torchaudio cudatoolkit=11.0 -c pytorch
完成安装后,您可以使用以下命令验证安装:
python -c "import torch; print(torch.__version__)"
如果一切顺利,您应该会看到PyTorch的版本信息。