ubuntu anaconda 安装pytorch
时间: 2025-01-20 08:55:12 浏览: 31
如何在 Ubuntu 上通过 Anaconda 安装 PyTorch
准备工作
确保已经成功安装了Anaconda或Miniconda。如果尚未完成这一步骤,可以参考官方文档或其他教程来获取最新的安装指南[^1]。
创建新的Conda环境
建议为不同的项目创建独立的Python环境以避免依赖冲突。可以通过下面命令建立一个新的环境:
conda create --name pytorch_env python=3.9
此操作会构建名为pytorch_env
的新环境,并指定Python版本为3.9。可以根据实际需求调整名称和Python版本号[^2]。
激活新创建的环境
使用如下指令激活刚才创建好的环境:
conda activate pytorch_env
一旦执行该命令后,在终端输入的所有后续命令都将在所选环境中运行[^3]。
安装PyTorch及相关工具包
对于CPU-only版本(不支持GPU加速),可以直接采用以下方法进行安装:
conda install pytorch torchvision torchaudio cpuonly -c pytorch
而对于希望启用CUDA支持的情况,则需根据具体的硬件配置选择合适的安装方式。例如针对CUDA 11.7的支持可参照下列语句:
conda install pytorch torchvision torchaudio cudatoolkit=11.7 -c pytorch
上述两条命令均来自官方推荐的方式之一,能够保证软件之间的兼容性[^4]。
验证安装是否成功
为了确认PyTorch已经被正确设置好,可以在Python解释器内部尝试导入库文件并打印其版本信息作为简单的测试手段:
import torch
print(torch.__version__)
如果一切正常的话,应该可以看到当前已加载的PyTorch的具体版本号码显示出来[^5]。
相关推荐


















