安装pytorch1.10.0+cuda11.4.3+cudnn8.2.1

入门三件套,安装的时候看自己需要哪个版本的,比如我用torchvision0.11.1,目前的最新版,适配pytorch1.10.0,安装的时候如果不安装cuda,就只能在cpu上训练模型。安装cuda的时候要先打开nvdia控制面板,左下角系统信息,组件,看一看自己的驱动支持的cuda版本,我这里是cuda11.4,适配的cudnn版本是8.2.1。

安装pytorch从清华镜像源

pip install torch===1.10.0 torchvision===0.11.1 -i https://pypi.tuna.tsinghua.edu.cn/simple

如果报错,则在-i前面加上-user

cuda可以到nvdia下载对应的版本,这是11.4版,注意网址里是.cn,如果是.com改成.cn这样下就快了,链接在下面。同时我把cuda11.4.3和cudnn8.2.1上传到了百度网盘

链接:https://pan.baidu.com/s/1Y8Ay4dOOdgahBRUQpy_1CA 
提取码:u2f7

cudnn需要登录nvdia开发者会员,但是登录界面被墙了,需要翻墙才能下载,需要下载其他版本的cudnn只能翻墙或者找了,https://developer.download.nvidia.cn/compute/cuda/11.4.3/local_installers/cuda_11.4.3_472.50_win10.exeicon-default.png?t=LA92https://developer.download.nvidia.cn/compute/cuda/11.4.3/local_installers/cuda_11.4.3_472.50_win10.exe

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值