Tensorflow-gpu – obaby@mars

Cannot open pip-script.py

2020年12月3日没有评论

这几天一直在折腾faceswap，安装过程中发现各种神奇的bug。首先第一个就是github无法正常访问的问题，clone代码的时候各种提示服务器连接超时。可以修改hosts文件添加以下内容：

# github
#192.30.255.113 github.com
#192.30.255.113 gist.github.com
#151.101.52.133  raw.githubusercontent.com
#151.101.53.194  github.global.ssl.fastly.net
140.82.114.4 github.com
199.232.69.194 github.global.ssl.fastly.net

添加之后执行dns刷新：

ipconfig /flushdns

然后github基本就可以正常访问了。安装完成之后发现出现了另外一个问题，提示没有tensorflow-gpu环境。但是尝试执行安装时又提示下面的错误：

人工智能『AI』

CUDNN_STATUS_NOT_INITIALIZED

2019年3月5日一条评论

自从装好tensorflow-gpu 之后其实一直没怎么用，今天跑代码的时候才发现安装的有问题：

测试代码如下：

from sklearn.datasets import load_sample_image
import matplotlib.pyplot as plt
import numpy as np
import tensorflow as tf

if __name__ == '__main__':
    # Load sample images
    china = load_sample_image("china.jpg")
    flower = load_sample_image("flower.jpg")
    dataset = np.array([china, flower], dtype=np.float32)
    batch_size, height, width, channels = dataset.shape
    # Create 2 filters
    filters = np.zeros(shape=(7, 7, channels, 2), dtype=np.float32)
    filters[:, 3, :, 0] = 1 # vertical line
    filters[3, :, :, 1] = 1 # horizontal line
    # Create a graph with input X plus a convolutional layer applying the 2 filters
    X = tf.placeholder(tf.float32, shape=(None, height, width, channels))
    convolution = tf.nn.conv2d(X, filters, strides=[1,2,2,1], padding="SAME")
    with tf.Session() as sess:
        output = sess.run(convolution, feed_dict={X: dataset})
    plt.imshow(output[0, :, :, 1], cmap="gray") # plot 1st image's 2nd feature map
    plt.show()

人工智能『AI』