Anaconda环境下的pytorch配置

本文详细介绍了如何通过Anaconda配置环境并安装PyTorch,包括设置清华镜像、创建虚拟环境、选择PyTorch及其相关组件版本、直接安装与手动安装的两种方法,以及在遇到问题时如何卸载和重新安装PyTorch。
摘要由CSDN通过智能技术生成

通过Anaconda来安装

  1. 配置好Anaconda环境

    1. 配置好清华镜像或者其他镜像
    2. 设置延时参数以应对网速慢的情况
    3. 新建虚拟环境
  2. 确定目标的pytorch版本

    • latest-version 或 previous-versions​​​​​​​中选择需要的pytorch版本
    • 同时确定其他重要组件的版本
      • torchvision
      • torchaudio
      • cudatoolkit-version

    下载地址:https://pytorch.org/get-started/locally/ or https://pytorch.org/get-started/previous-versions/

  3. 安装方式

    切换到对应的虚拟环境

    source/conda activate {env-name}
    
    1. 第一种,直接通过torch官方指导的命令行进行直接安装,conda会直接指定适配的版本。

      conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
      

      1. 网速稳定快速,能够顺利安装
      2. 该命令会将所有需要的库下载到路径:~/anaconda3/pkgs下,并逐一进行安装
    2. 手动本地安装

      1. 特殊版本的下载,通过Chrome或wget命令

        1. 清华镜像的torch库,下载pytorch,torchvision,torchaudio

        2. cudatoolkit下载 -- 版本与CUDA Runtime Version对应

        • 这些大文件在下载之前可以去访问一下服务器的公共资源库,看一下有没有,可以避免重复下载.
      2. 手动安装大文件

        conda install --offline {root}/{pkg-name} # 
        

        成功解压和安装,会将解压的文件保存在~/anaconda3/pkgs/{pkg-name}下

      3. 剩余文件安装

        1. 使用Pytorch官网推荐的命令,如

          conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
          
  4. 卸载与重新安装

    如果遇到pytorch版本安装错误时,可以使用以下命令卸载已安装的pytorch

    source activate {env-name}
    conda uninstall pytorch 
    

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值