- 首先,添加清华镜像源,添加指令如下:
#添加镜像源
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/pro
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2
#显示检索路径
conda config --set show_channel_urls yes
#显示镜像通道
conda config --show channels
- 创建tensorflow-gpu1.13.1虚拟环境:
conda create -n [name] python==3.7.3 tensorflow-gpu==1.13.1
- 安装成功后,通过activate [name] 激活虚拟环境;
- 测试是否安装成功:
import tensorflow as tf
test = tf.constant('Success!')
sess = tf.Session()
print(sess.run(test))
运行结果:
2021-01-16 13:33:16.728119: I tensorflow/core/platform/cpu_feature_guard.cc:141] Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX AVX2
2021-01-16 13:33:17.413233: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1433] Found device 0 with properties:
name: GeForce MX230 major: 6 minor: 1 memoryClockRate(GHz): 1.531
pciBusID: 0000:01:00.0
totalMemory: 2.00GiB freeMemory: 1.63GiB
2021-01-16 13:33:17.413630: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1512] Adding visible gpu devices: 0
2021-01-16 13:33:21.247634: I tensorflow/core/common_runtime/gpu/gpu_device.cc:984] Device interconnect StreamExecutor with strength 1 edge matrix:
2021-01-16 13:33:21.247818: I tensorflow/core/common_runtime/gpu/gpu_device.cc:990] 0
2021-01-16 13:33:21.247954: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1003] 0: N
2021-01-16 13:33:21.248223: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1115] Created TensorFlow device (/job:localhost/replica:0/task:0/device:GPU:0 with 1369 MB memory) -> physical GPU (device: 0, name: GeForce MX230, pci bus id: 0000:01:00.0, compute capability: 6.1)
b'Success!'
Process finished with exit code 0