因为要看一下别人的代码怎么样,我平时用的是pytorch,所以开了个虚拟环境安装tensorflow及其GPU版本。
我大概看了下,tensorflow最后更新的1.x的版本是1.15的。所以我安装的是1.15.x的版本。
虚拟环境相当于重新开一个文件夹,在这个文件夹里搭建好python的包,不影响自己平时用的python环境,非常方便。虚拟环境的安装我就不多说,大家可以去查阅其他教程。
我之前因为pytorch已经安装好了cuda+cudnn 11.1的版本,但是tensorflow==1.15.x只支持到10.0版本。所以我们后面还需要修改一些文件,来符合要求。
安装
虚拟环境中,开启activate后,在命令行
pip install tensorflow==1.15.3
#版本随便选的
接着
pip install tensorflow-gpu==1.15.3
检测开启gpu
在命令行输入:python之后,或者在pycharm等编辑器里面运行下面的代码:
import tensorflow as tf
print(tf.test.is_gpu_available())
输出结果: