Anoconda虚拟环境安装Pytorch(GPU版本)

文章介绍了如何在Anaconda虚拟环境中通过whl文件安装PyTorch的GPU版本,强调了版本匹配的重要性,包括PyTorch版本、Python版本和CUDA版本,并提供了命名规则的解释,以及详细安装步骤。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

在这里插入图片描述

在用虚拟环境安装PyTorch的时候常常因为HTTPerror等大段地红字错误所烦恼,即使安装成功也会因为Python和PyTorch版本不匹配等问题阻碍我们下一步的开发。So我建议使用whl文件进行安装,可以减少以上麻烦的发生。

版本选择

Pytorch的官方whl文件下载地址
点击链接就可以看到一个链接汇总页面,向下翻(不介绍链接名中含有cpu的链接)

在这里插入图片描述
torch-0.4.1-cp35-cp35m-linux_x86_64.whl
torch-0.4.1-cp35-cp35m-win_amd64.whl
torch-1.5.0-cp38-cp38-linux_x86_64.whl
torch-1.5.0-cp38-cp38-win_amd64.whl

以以上几个whl链接名为例,介绍一下他的命名规则,方便进一步下载

  1. torch:PyTorch是一个基于Torch的Python开源机器学习库,所以下载对应的torch依赖才能正常使用Pytorch
  2. 0.4.1/1.5.0对应着需要安装的PyTorch版本,注意PyTorch版本的选择还需要考虑PC中CUDA版本
    PyTorch和CUDA对应版本如下,高版本的CUDA环境可以向下兼容(10.2和11.3的CUDA能兼容大部分版本的PyTorch,具体以电脑实际情况为准,自己用的电脑CUDA是12.1的,用1.9版本的PyTorch依然没有问题
    PyTorch
  3. cpXX表示Python的版本,cp35对应Python3.5版本,cp38对应Python3.8版本,以此类推
  4. win_amd64/linux_x86_64表示安装环境,linux_x86_64表示需要安装在linux系统下,win_amd64表示安装在Windows系统下

*在下载界面下面还会出现类似
torch-1.8.1+cu102-cp36-cp36m-linux_x86_64.whl
torch-1.8.1+cu102-cp36-cp36m-win_amd64.whl
这样的链接,这种就要求更为严格一些,需要CUDA和PyTorch一一对应才能正常使用
cu102即CUDA10.2版本。
*
以上就是命名规则的介绍,根据项目需求,开发环境,按需选择即可

选择链接并点击即可开始下载

安装流程

  1. 复制下载好的whl文件的路径(右键-属性-安全)
    在这里插入图片描述
    复制链接
  2. 打开Anaconda Prompt,建议使用管理员模式运行。
  3. 切换到指定的虚拟环境(这里不介绍),输入
pip install whl文件路径

回车,

即可开始安装
按照安装提示有序进行下一步就可以了

以上就是GPU版本PyTorch的安装教程,自认为用whl安装可以减少很多不必要的麻烦,如果有误请多多指教!

### 安装适用于GPUPyTorch #### 创建Anaconda虚拟环境 为了确保兼容性和性能,在安装PyTorch-GPU之前应当先创建一个新的Conda虚拟环境。这一步骤可以隔离不同项目的依赖关系,防止冲突发生。 ```bash conda create -n pytorch_env python=3.9 ``` 此命令会建立名为`pytorch_env`的新环境,并设置Python版本为3.9[^2]。 #### 激活新创建的虚拟环境 一旦完成环境创建,则需激活该环境以便在其内部执行后续操作: ```bash conda activate pytorch_env ``` 通过这条指令可切换至先前定义好的工作空间内继续下一步的工作流程[^1]。 #### 查找合适的PyTorch版本组合 考虑到CUDA驱动程序与特定版本之间的匹配度问题,建议访问官方文档获取最新的稳定版推荐列表来决定具体要使用的PyTorch、CUDA以及其他组件的具体版本号[^3]。 #### 使用Conda安装PyTorch GPU版本 确认所需参数无误之后,可以通过下面的方式一次性完成所有必要的软件包部署: ```bash conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch ``` 这里假设选择了CUDA Toolkit 11.3作为目标计算平台的支持库;实际应用时应依据个人硬件条件调整相应选项。 #### 验证安装成果 最后但同样重要的是验证整个过程是否顺利完成。可以在交互式解释器中尝试导入模块并查看设备状态来进行简单的测试: ```python import torch print(torch.cuda.is_available()) ``` 如果返回True则说明已经成功启用了GPU加速功能。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值