Torch 、torchvision 、Python 版本对应关系以及安装 GPU 或 CPU 版本的 pytorch

1. torch - torchvision - python 版本对应关系

来自pytorch官网
从表中可以看出,在使用 anaconda 创建虚拟环境时,将 python 的版本定为 3.7 最为合适,当然最好还是根据你自己的需要选择 python 版本。

conda create -n 环境的名字 python=3.7

2. CUDA Toolkit 和PyTorch的对应关系

仅供参考

3. 安装说明

3.1 用 anaconda 安装 pytorch

anaconda 新建虚拟环境后,直接在 pytorch 官网官网链接 找到 “Install” 按钮。这里一键搞定torch,torchvision,cudatoolkit 等等。有以下几点需要注意和知晓的:

  • 前提需要安装好 NVIDIA 驱动 。

  • 不需要另外安装 CUDA(我在没有单独安装 CUDA 情况下,功运行了 torch-gpu )

  • 与电脑或者服务器上已经装好的 CUDA 不会发生冲突 (这一点相比安装 tensorflow 要友好得多)。

  • anaconda 会自动对应匹配 torch,torchvision,cudatoolkit 等的版本。不过在创建虚拟环境的时候要选择合适的 python 版本 。

  • 如果不指定 torch 版本,这里一般下载最新的 torch 版本 。

  • anaconda 将 torch、torchvision、cudatoolkit 、python 等等依赖包都集成在创建的虚拟环境里,统一管理依赖包,匹配各自之间的版本。

有一点是需要特别注意的,但是在你们安装时,不一定会碰得上。在不指定版本的情况下,anaconda 会下载最新能匹配的依赖包版本 ,但不一定是你需要的版本;在指定下载版本的情况下,如果用 conda install xxxx=verision1 , anaconda 中如果没有版本为 version1 的 xxxx 依赖包,则会直接报错,如果是使用 pip install XXX==version2,即使是 anaconda 中没有版本为 version2 的 XXX 依赖包,anaconda 也不会报错,并且使用命令 conda list 会显示安装了 version2 版本的 XXX 依赖包,但是实际上并没有安装好该依赖包 。

  • 84
    点赞
  • 506
    收藏
    觉得还不错? 一键收藏
  • 3
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值