栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

【记录-自用】Anoconda安装Pytorch-GPU踩坑点

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

【记录-自用】Anoconda安装Pytorch-GPU踩坑点

目录
  • 1 下载和安装CUDA和Cudnn
  • 2 安装Pytorch-GPU
  • 3 验证是否安装成功

前提:已经安装了Anoconda

1 下载和安装CUDA和Cudnn

第一次装,参考这篇博客: 下载和安装CUDA和Cudnn(图文详解)
第二次装,想先卸载以前的cuda,参考这篇博客:windows下CUDA的卸载以及安装

踩坑点:

  • 官网下载Cudnn需要注册账号后才能下载,建议外网注册。
2 安装Pytorch-GPU

Pytorch官网安装方法失败,去掉"-c pytorch"安装出来的pytorch和torchvision版本对应失败。
我的方法: 离线下载pytorch、torchvision 点击这里下载,下载时注意cuda、python的版本,使用命令"pip debug --verbose"查看python,再执行pip install指令。一定要是对应版本,见踩坑点最后一点。

【PS】:https://download.pytorch.org/whl/torch_stable.html这个网站有时候打不开,真奇怪。

踩坑点:

  • torch命令行执行pip install 不要用镜像下载,否则仍然是cpu版本。
  • 如果已经安装了cpu版本的pytorch,建议重新创建一个虚拟环境安装gpu版本的。
  • pytorch、torchvision 对应版本:Github网站
3 验证是否安装成功
import torch
print(torch.__version__)
print(torch.cuda.is_available())
转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/958076.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号