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文件,去清华源官网,把一整段话直接复制进去就行
pytorch-gpu环境安装,适用于深度学习,yolov3_第1张图片
③进入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

你可能感兴趣的:(环境安装)