安装PyTorch在Linux系统上通常需要执行以下步骤:
conda create -n pytorch_env python=3.7
conda activate pytorch_env
conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch
如果不需要CUDA支持,可以使用以下命令安装CPU版本:
conda install pytorch torchvision torchaudio cpuonly -c pytorch
import torch
print(torch.__version__)
通过以上步骤,您应该能够成功在Linux系统上安装PyTorch。如果遇到任何问题,建议查看PyTorch官方文档或社区论坛上的相关帮助。
亿速云「云服务器」,即开即用、新一代英特尔至强铂金CPU、三副本存储NVMe SSD云盘,价格低至29元/月。点击查看>>
推荐阅读:pytorch安装的步骤是什么