pytorch-gpu环境安装,适用于深度学习,yolov3

本文采用的配置win10+torch1.5
①首先需要安装cuda+cudnn,考虑到cuda和cudnn版本兼容问题,还有官网下载速度极慢,直接csdn搜索cuda10.2 cudnn,去某宝用券下载就行

····测试是否安装成功cmd下输入命令nvcc-V
····cudnn压缩包解压出的三个文件夹里的文件,需要分别复制到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2对应的文件夹中。
····然后配置环境变量,在path中添加C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\bin
在这里插入图片描述
②修改anaconda清华源,直接在C盘用户里找到.condarc文件,去清华源官网,把一整段话直接复制进去就行
在这里插入图片描述
③进入anaconda prompt
首先创建虚拟环境:conda create -n [自定义环境名字] python=3.7
然后激活环境:conda activate [自定义环境名字]
最后安装pytorch1.5:conda install pytorch torchvision cudatoolkit=10.2 -c pytorch
切记最后还要安装cudnn,版主就是因为没安装这个困扰了一天:conda install cudnn

到这里基本就结束了,以下是自定义安装

①apex安装:https://blog.csdn.net/weixin_44529105/article/details/106143420
②pycocotools安装:https://blog.csdn.net/weixin_44529105/article/details/106134280
③opencv安装:https://blog.csdn.net/weixin_44529105/article/details/106128588

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值