在Linux上安装PyTorch的步骤如下:
conda --version
,若显示版本号(如 conda 23.7.4
),则安装成功。conda create -n pytorch_env python=3.8
Python 3.8是兼容PyTorch的稳定版本,可替换为3.9或3.10(需确认PyTorch支持)。
conda activate pytorch_env
激活后终端提示符应显示 (pytorch_env)
。
conda install pytorch torchvision torchaudio cpuonly -c pytorch
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
nvidia-smi
在创建的小环境下输入 python
,在Python解释器中输入以下代码,返回 True
,安装成功:
import torch
torch.cuda.is_available()
以上步骤应该可以帮助你在Linux系统上成功安装PyTorch。如果在执行过程中遇到问题,可以参考PyTorch的官方文档或相关社区论坛寻求帮助。
辰迅云「云服务器」,即开即用、新一代英特尔至强铂金CPU、三副本存储NVMe SSD云盘,价格低至29元/月。点击查看>>
推荐阅读: linux如何查看文件系统剩余空间