为什么不尝试Anaconda?
conda create -n gpu_env tensorflow-gpu
conda activate gpu_env
就这么简单。当您敬畏观看时,所有需要的软件包和运行时(包括Visual Studio)将被放在一起。 试试看!
警告:您需要从tensorflow内部运行Keras ,以使其检测和使用GPU 。否则,如果仅使用keras和tensorflow作为后端, 那么Keras将使用普通的tensorflow而不是tensorflow-gpu 。两者并排安装。
意思是您应该这样编码:
import tensorflow as tf
from tensorflow.keras.models import Sequential, load_model
from tensorflow.keras.layers import Dense,Dropout
from tensorflow.keras.callbacks import ModelCheckpoint, EarlyStopping,CSVLogger
from tensorflow.keras.utils import plot_model
.............
.............
希望能有所帮助。
0
我正在尝试在PC上安装Tensorflow,我安装了适用于Windows 10的Python 3.5.2 64位,cuda_8.0.61和cudnn-8.0-windows10-x64-v6.0
我使用“本地” pip安装了Tensorflow的GPU版本,然后打开了IDLE,并使用“将tensorflow导入为tf”进行了测试,但出现了以下错误
我怎样才能解决这个问题 ?预先感谢您<3