在CentOS上安装PyTorch的依赖项,通常推荐使用conda来管理,因为它可以自动处理依赖关系并确保环境的一致性。以下是使用conda安装PyTorch依赖项的步骤:
conda create -n torch_env python=3.8
conda activate torch_env
# 安装PyTorch和torchvision,指定CUDA版本(以CUDA 11.3为例)
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
import torch
print(torch.__version__)
print(torch.cuda.is_available())
请注意,具体的依赖项和版本可能会随着PyTorch的更新而变化,因此建议参考PyTorch官方文档或GitHub页面上的最新安装指南来获取最准确的信息。
亿速云「云服务器」,即开即用、新一代英特尔至强铂金CPU、三副本存储NVMe SSD云盘,价格低至29元/月。点击查看>>