标签: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. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。