Win10 ancona傻瓜安装tensorflow-gpu,ancona傻瓜安装pytorch-gpu

楔子

之前是在ubuntu16.04LTS上安装的是python2.7非anconda方式的tensorflow-gpu。
1、学习使用一段时间后python2.7有点坑,网上的好多代码不能直接运行,比如:无法使用**表示keywords dic;
2、而且在ubuntu使用spyder始终无法实现汉字的输入,对于学习阶段注释很重要,很不方便;
3、后来偶然使用wintogo制作了一个win10的系统,就像基于anconda方式傻瓜式搭建。

最后没有能用conda的傻瓜装法,还是在全局下安装CUDA和cuDNN,这样版本才能可控。

anconda的安装:

直接官网下载,傻瓜式安装。(环境变量可以勾,不勾就需要自己配置)
https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/Anaconda3-5.3.1-Windows-x86_64.exe

CUDA,cuDNN的安装:

注意了他们两个是配套的,先下CUDA:
在这里插入图片描述
再找对应版本的cudnn
先安装CUDA,再把cuDNN解压,把这三个文件夹的文件放进CUDA安装目录的三个相同的文件夹内:
在这里插入图片描述
cuDNN是对CUDA的补充,是针对DNN加速的,就这么理解。
手动添加一下如下环境变量,明显他是针对cuDNN的环境变量

在这里插入图片描述

tensorflow-gpu的安装:

配置一个tensorflow的环境:

如下就是配置一个python版本3.6的环境,环境的名字叫tensorflow :

(base) C:\windows\system32>conda create -n tensorflow pip python=3.6

为什么需要专门配置一个新的环境,anconda有一个默认的环境base,理论上可以安装在默认环境下,但是为了防止依赖冲突,最好是新建一个环境,实际上在base下直接安装会报错:

(base) C:\windows\system32>conda install tensorflow-gpu
Solving environment: failed

UnsatisfiableError: The following specifications were found to be in conflict:
  - tensorflow-gpu
Use "conda info <package>" to see the dependencies for each package.

进入新的环境:

(base) C:\windows\system32> conda activate tensorflow
(tensorflow) C:\windows\system32>
(tensorflow) C:\windows\system32>

安装tensorflow-gpu:

建议不要使用conda install tensorflow-gpu,他会下载使用自己的CUDA,之前说的傻瓜就是说的这个,关键是它自己CUDA和tensorflow-gpu不匹配使用起来存在问题,至少我试过python3.5和3.6都有问题。
使用pip安装指定版本。

pip install --ignore-installed --upgrade tensorflow_gpu==1.8.0

安装之前最好把pip更新到最新版本;
按照过程会安装很多依赖,需要等待;
还有tensorflow_gpu版本需要特别注意,要和CUDA,cuDNN配套,最新版本说不定就不支持,这里给一个过时的支持:
在这里插入图片描述
可以直接在创建的容器里使用python测试:
在这里插入图片描述

import tensorflow as tf
import numpy as np
import time
 
# 产生用于计算测试的数据
value = np.random.randn(5000,1000)
a = tf.constant(value)
 
# 计算方式
b = a*a
 
# gpu
tic = time.time()
with tf.Session() as sess:
    for i in range(1000):
        sess.run(b)
toc = time.time()
t_cost = toc - tic
 
print(t_cost)

安装spyder:

在容器里安装spyder:
在这里插入图片描述
如果base环境里面有spyder, 在这个环境里会直接copy过来的
还有一点在spyder里面看不见使用GPU的日志,linux里面是可以看到的,不管了反正使用的是GPU计算就是了。

pytorch安装:

CUDA和cuDNN已经全局安装好了:
在这里插入图片描述

pip3 install https://download.pytorch.org/whl/cu90/torch-1.0.1-cp36-cp36m-win_amd64.whl
pip3 install torchvision

假如外部没有安装CUDA和cuDNN,可以试试:
在这里插入图片描述
成功结果如下:
在这里插入图片描述

conda commands:

# 创建新环境
conda create -n rcnn python=3.6

# 删除环境
conda remove -n rcnn --all

# 重命名环境
conda 其实没有重命名指令,实现重命名是通过 clone 完成的,分两步:

# 先 clone 一份 new name 的环境
# 删除 old name 的环境

# 比如,想把环境 rcnn 重命名成 tf
# 第1步
conda create -n tf --clone rcnn

Source:      /anaconda3/envs/rcnn
Destination: /anaconda3/envs/tf
Packages: 37
Files: 8463

# 第2步
conda remove -n rcnn --all

# 结果
conda info -e
# conda environments:
#
crawl                    /anaconda3/envs/crawl
flask                    /anaconda3/envs/flask
tf                       /anaconda3/envs/tf
root                  *  /anaconda3

猜你喜欢

转载自blog.csdn.net/weixin_40759186/article/details/87458006