环境配置 pytroch 、keras框架

环境配置 pytroch 、keras框架,第1张

环境配置:先安装Anaconda,再安装cuda,配置环境
一、Anaconda安装

Anaconda 方便管理不同环境,放置不同框架:pytorch、tensorflow、keras..                                  

1、Anaconda的下载官网:https://www.anaconda.com/distribution/  ,下载64位安装包。

2、Anaconda的安装位置D/E/F...,一般都不放C盘,懂得都懂。

选择Add Anaconda to my PATH environment variable,自动将anaconda装到系统的环境变量。

二、Cudnn和CUDA的下载和安装

目前网络常用torch=1.2.0,官方推荐的Cuda版本是10.0,对应的cudnn是7.4.1。

1、Cudnn和CUDA的官网下载,或者找网盘...

2、Cudnn和CUDA的安装
注意:安装前关闭360安全管家等,防止 xxx.dll 文件拒绝访问,下载好用管理员方法打开exe安装

根目录 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0  对应的解压lib文件覆盖...  


三、配置torch环境

1、pytorch环境的创建与激活,  Anaconda Prompt 命令:

conda create –n pytorch-a python=3.6     创建一个名为pytorch-a的环境,该环境python版本3.6。
activate pytorch-a                                     激活一个名为pytorch-a的环境。

2、pytorch库的安装.  激活环境内下载,注意在anaconda 路径转换下,cd D/:.../
a、官方推荐安装方法(推荐): https://pytorch.org/get-started/previous-versions/
官网推荐的安装Cuda10的版本,pytorch官方提供的指令,用于安装torch和torchvision

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

b、先下载whl后安装,这样安装特别慢,网址:https://download.pytorch.org/whl/torch_stable.html
找到对应版本下载  ,完成后找到安装路径,进行覆盖...

3、深度学习模型其它依赖库的安装,具体如下:

scipy==1.2.1
numpy==1.17.0
matplotlib==3.1.2
opencv_python==4.1.2.30
torch==1.2.0
torchvision==0.4.0
tqdm==4.60.0
Pillow==8.2.0
h5py==2.10.0

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple 清华镜像源

tensorboard
future

四 keras环境

1.    conda create -n keras #创建虚拟环境
2.    conda activate keras #进入虚拟环境

3.    pip install tensorflow-gpu==1.15 -i http://pypi.douban.com/simple --trusted-host pypi.douban.com  豆瓣镜像
4.    pip install keras==2.3.1 -i http://pypi.douban.com/simple --trusted-host pypi.douban.com

注意tensorflow和keras对应版本

其他包:

5. pip install scikit-image -i http://pypi.douban.com/simple --trusted-host pypi.douban.com

pip install pyparsing==2.4.7 -i http://pypi.douban.com/simple --trusted-host pypi.douban.com

欢迎分享,转载请注明来源:内存溢出

原文地址: http://outofmemory.cn/zaji/957003.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2022-05-18
下一篇 2022-05-18

发表评论

登录后才能评论

评论列表(0条)

保存