【pytorch】安装pytorch的步骤与问题记录

本文详细记录了PyTorch GPU版本在Windows和Linux环境下的安装步骤,包括显卡驱动、CUDA、cuDNN的安装,并提供了相关资源链接。遇到驱动、CUDA版本匹配问题时,无需重复安装,除非需要更换版本。当conda或pip安装指定版本PyTorch失败时,可能默认安装CPU版本。
摘要由CSDN通过智能技术生成

装了很多次pytorch了,但隔段时间总会遗忘或遇到些问题,此处记录一下。

安装步骤

对于安装流程很多人都写过相关blog,此处不在赘述。下面贴一下参考链接:

  1. Windows安装Pytorch-GPU版本
  2. PyTorch-GPU 安装之 conda install & pip install 区别详解
  3. 服务器RTX3090安装显卡驱动,cuda,cudnn,gpu版pytorch记录
  4. Linux安装Pytorch1.8GPU(CUDA11.1)的实现

常用网址

  1. torch,torchvision离线下载地址
  2. torch, torchvision, python 的版本对应关系
  3. cuda与显卡驱动的 版本对应关系
  4. NVIDIA显卡驱动下载地址
  5. cuda下载地址
  6. cudnn下载地址
  7. pytorch官网

疑问解答

  1. 显卡驱动, cuda, cudnn的关系与版本对应问题
  2. 本机已有显卡驱动和cuda,直接装pytorch即可。不必在重新装驱动和cuda,除非你想换其他版本的驱动。
  3. pip安装
  4. conda或pip装 pytorch指定版本时,若服务器端找不到对应可下载的库,则会默认安装CPU版本的torch
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值