ananconda安装pytorch教程
时间: 2025-02-22 07:37:29 浏览: 39
使用 Anaconda 安装 PyTorch 的教程
创建 Python 虚拟环境
为了确保不同项目之间的依赖项不会冲突,建议为每个项目创建独立的虚拟环境。对于安装 PyTorch 来说也不例外。
conda create -n pytorch python=3.10
这条命令会基于指定的 Python 版本 (这里是 3.10) 创建一个新的名为 pytorch
的 Conda 环境[^2]。
激活新创建的环境并安装 PyTorch
一旦环境被成功创建之后,激活该环境:
conda activate pytorch
接着可以按照官方推荐的方式通过 Conda 渠道来安装带有 CUDA 支持的 PyTorch GPU 版本。假设计算机上安装有 NVIDIA 显卡,并且支持 CUDA 11.8 或更低版本,则可执行如下命令完成安装:
conda install pytorch torchvision torchaudio cudatoolkit=11.8 -c pytorch
这里特别需要注意的是 cudatoolkit
参数后面的数值应该匹配实际硬件所能够支持的最大 CUDA 版本号;如果不确定具体的支持情况可以通过运行 nvidia-smi
命令查看显卡信息以及其兼容的 CUDA 版本范围[^3]。
验证安装是否成功
最后一步是要验证刚刚安装好的库能否正常工作,在终端里启动 Python 解释器尝试导入 PyTorch 库来进行简单的测试:
import torch
print(torch.cuda.is_available())
这段代码将会打印出 True 表明当前环境中已经正确配置好了具有 GPU 加速功能的 PyTorch 实例。
相关推荐















