当前位置 博文首页 > Weiyaner的博客:anaconda安装tenforflow-gpu(以我的mx250显卡
今天手贱,在anaconda navigator里面,看到了有重复的TF2.1文件夹,我以为是以前自己命名重复了好几个tensorflow环境,就手贱的点了remove,万万没想到,一个remove让所有的TF2.1都没了,我以前辛辛苦苦安装的tensorflow-gpu版本也一并消失。所以写一个帖子记录下再次安装成功的过程,防止下次手贱。
官网:点击这里
已安装的请忽略即可
三者的版本号有特定要求:查看以下网址:
https://www.tensorflow.org/install/source_windows 或者 点击这里
下载网址:https://developer.nvidia.com/cuda-toolkit-archive 或者点击这里
进入anaconda promt里面
1.首先创建一个文件夹,配置好python环境
conda create -n TF2.1 python=3.7
2.激活环境
conda activate TF2.1
3.下载tensorflow-gpu
从豆瓣镜像下载所需要的版本
conda install tensorflow-gpu=2.1.0 -i https:pypi.douban.com/simple
进入python环境,输入代码:
import tensorflow as tf
tf.__version__
测试gpu情况:
import tensorflow as tf
print("Num GPUs Available: ", len(tf.config.experimental.list_physical_devices('GPU')))
大功告成。