anaconda安装tensorflow-gpu注意事项

1 查询cuda与anaconda中的python版本(已添加环境变量)
在这里插入图片描述
在这里插入图片描述
2 查询cuda和python对应的tensorflow-gpu版本
下载网址:https://www.tensorflow.org/install/pip?lang=python2
我下载的版本:
tensorflow_gpu-2.6.0-cp38-cp38-win_amd64.whl

3 安装(可以新建虚拟环境安装,也可以在ite-packages中使用pip install安装,本文选择后者)
将下载的tensorflow_gpu-2.6.0-cp38-cp38-win_amd64.whl拷贝至Anaconda3\Lib\site-packages中,进入该文件夹中,输入命令pip install tensorflow_gpu-2.6.0-cp38-cp38-win_amd64.whl,出现错误:
在这里插入图片描述
解决方案:
pip install –user tensorflow_gpu-2.6.0-cp38-cp38-win_amd64.whl
在这里插入图片描述
4 测试代码
import tensorflow as tf
print(tf.version)
print(‘GPU’,tf.config.list_physical_devices(‘GPU’))
在这里插入图片描述

文章出处登录后可见!

已经登录?立即刷新

共计人评分,平均

到目前为止还没有投票!成为第一位评论此文章。

(0)
心中带点小风骚的头像心中带点小风骚普通用户
上一篇 2022年4月1日 下午10:29
下一篇 2022年4月1日 下午10:39

相关推荐