找回密码
 会员注册
查看: 27|回复: 0

python3.7安装并配置pytorch(torch1.8.2+cuda11.1+torchaudio0.8.2+torchvision0.9.2)

[复制链接]

2万

主题

0

回帖

6万

积分

超级版主

积分
65944
发表于 2024-9-4 18:06:02 | 显示全部楼层 |阅读模式
文章目录前言一、安装python二、安装cuda+cudnn二、安装pytorch2.1版本匹配2.1.1方法一2.1.2方法二2.2安装.tar.bz2三、验证是否安装成功总结前言本篇文章主要介绍在Windows下python3.7配置pytorch,帮助需要的朋友避坑安装pytorch需要多个版本适配,本文提供一种使用于python3.7和cuda的安装方法,同时给出一些处理问题的建议一、安装pythonpython3.7是比较稳定的版本,可以根据自己的需求安装,可以参考博客:anaconda安装补充:anaconda历史版本仓库二、安装cuda+cudnn参考安装博客:cuda安装补充:查看cuda版本nvidia-smi只要下载的cuda版本不高于自己显卡支持的版本就行二、安装pytorch2.1版本匹配根据自己安装的cuda去找对应的pytorch、torchvision和torchaudio2.1.1方法一去pytorch官网找对应的版本:官网地址需要注意的是:torch、torchvision以及torchaudio版本一定要对应,具体看官网给出的版本。比如,cuda11.8可以安装torch2.0.1+torchvision0.15.2+torchaudio2.0.2在换源后复制安装命令进行安装,如这里的cuda11.8如果顺利的话,这一行命令就可以解决问题,但是可能因为版本不兼容等问题,这里出现的版本可能在仓库中不存在。比如我要安装的python3.7+cuda11.1所有给出满足的版本如下:cuda11.1+pytorch仓库地址仓库地址大家可以根据自己要的cuda版本找,如下:一般而言在cmd中无法下载并报找不到对应版本错误,那么仓库中就没有2.1.2方法二由于python3.7+cuda11.1在上面所有版本中都出现了仓库中找不到某一个库的适配版本,比如能找到torch1.10.1,但是找不到torchvision0.11.2,意味着无法安装。解决办法卸载python3.7,安装python3.8卸载cuda11.1,装一个低版本都不用卸载,找清华镜像或者国内其他镜像,看是否有适合的版本torch的清华镜像补充:cu102是指cuda10.2py37是指python3.7cudnn8_0是指cudnn8.*2.2安装.tar.bz2可以将下载的.tar.bz2文件放在anaconda的pkgs下,并在该目录下打开命令行,安装命令:condainstall--use-local包名三、验证是否安装成功cmd进入python环境importtorchprint(torch.cuda.is_available())12总结运气不好装torch真的很需要耐心~~~~
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 会员注册

本版积分规则

QQ|手机版|心飞设计-版权所有:微度网络信息技术服务中心 ( 鲁ICP备17032091号-12 )|网站地图

GMT+8, 2024-12-31 04:09 , Processed in 1.322905 second(s), 26 queries .

Powered by Discuz! X3.5

© 2001-2024 Discuz! Team.

快速回复 返回顶部 返回列表