ICode9

精准搜索请尝试: 精确搜索
首页 > 编程语言> 文章详细

PyTorch搭载的CenterNet算法环境配置

2021-07-08 22:03:14  阅读:266  来源: 互联网

标签:https 1.2 torch pytorch CenterNet 算法 PyTorch org 官网


PyTorch搭载的CenterNet算法环境配置

环境内容

torch:1.2.0
torchvision:0.4.0


Anaconda安装

以前已安装

旧版链接: https://pan.baidu.com/s/12tW0Oad_Tqn7jNs8RNkvFA
提取码: i83n

网上搜索Anaconda的官网:https://www.anaconda.com/distribution/

在这里插入图片描述


下载Cudnn和CUDA

我这里使用的是torch=1.2.0,官方推荐的Cuda版本是10.0,因此会用到cuda10.0,与cuda10.0对应的cudnn是7.4.1.5,这个组合我实验过了,绝对是可以用的。

cuda10.0官网的地址是:
cuda10.0官网地址
cudnn官网的地址是:(7.4.1.5)
cudnn官网地址

百度云链接: https://pan.baidu.com/s/1znYSRDtLNFLufAuItOeoyQ
提取码: 8ggr

下载完之后得到这两个文件。

在这里插入图片描述

在这里插入图片描述

下载好之后可以打开cuda_10的exe文件进行安装。

在这里插入图片描述

选择自定义。

不

直接点下一步。

在这里插入图片描述

安装完后在C盘这个位置可以找到根目录。
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0

然后把Cudnn的内容解压。

在这里插入图片描述

把这里面的内容直接复制到C盘的根目录下就可以了。

在这里插入图片描述


配置torch环境

Win+R启动cmd,在命令提示符内输入以下命令:
创建环境:

conda create –n pytorch python=3.6

激活环境:

activate pytorch 

打开pytorch的官方安装方法:
https://pytorch.org/get-started/previous-versions/
官网推荐的安装代码如下,我使用的是Cuda10的版本:

# CUDA 10.0
pip install torch===1.2.0 torchvision===0.4.0 -f https://download.pytorch.org/whl/torch_stable.html

# CUDA 9.2
pip install torch==1.2.0+cu92 torchvision==0.4.0+cu92 -f https://download.pytorch.org/whl/torch_stable.html

# CPU only
pip install torch==1.2.0+cpu torchvision==0.4.0+cpu -f https://download.pytorch.org/whl/torch_stable.html

轮子链接:

https://download.pytorch.org/whl/torch_stable.html

在这里插入图片描述

在这里插入图片描述

在这里插入图片描述

百度云链接: https://pan.baidu.com/s/14-QVk7Kb_CVwaVZxVPIgtw
提取码: rg2e

标签:https,1.2,torch,pytorch,CenterNet,算法,PyTorch,org,官网
来源: https://www.cnblogs.com/cj-whales/p/14988348.html

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有