搞了一天才弄好,记录一下打开Anaconda Prompt(管理员模式打开)
(base) C:\Windows\system32>conda create --name tf2x python=3.8
(base) C:\Windows\system32>conda activate tf2x
(tf2x) C:\Windows\system32>conda install cudatoolkit=11.3 cudnn=8.2.1
(base) C:\Windows\system32>conda activate tf2x
(tf2x) C:\Windows\system32>pip install tensorflow-gpu==2.4.1 keras==2.4.3 -i https://pypi.douban.com/simple/
(这个图太长了 就不截那么多了)
必要的库 (以后安装库要用管理员)
(tf2x) C:\Windows\system32>pip install opencv-python==4.4.0.44 pillow==8.2 numpy==1.21 matplotlib scipy pandas scikit-learn tqdm imutils PyYAML seaborn protobuf==3.20 chardet cchardet -i https://pypi.douban.com/simple/
Looking in indexes: https://pypi.douban.com/simple/
(这个图也好长 不截那么多了)
下载了cusolver64_10.dll (我的系统缺少这个文件)
复制在Windows->System32下面
测试tensorflow有没有安装成功
import tensorflow as tf
import keras
print(tf.test.is_gpu_available())
print(tf.__version__)
print(keras.__version__)