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’))
文章出处登录后可见!
已经登录?立即刷新