创建一个新环境conda create --name tf_gpu python=3.6
,python版本为3.6,然后分别执行以下语句,conda install cupy
可以自动寻找符合版本的cuda和cudnn,一行代码即可完成,不需要自己下载。
conda install tensorflow-gpu=2.0.0
conda install cupy
conda install keras
创建一个新环境conda create --name tf_gpu python=3.6
,python版本为3.6,然后分别执行以下语句,conda install cupy
可以自动寻找符合版本的cuda和cudnn,一行代码即可完成,不需要自己下载。
conda install tensorflow-gpu=2.0.0
conda install cupy
conda install keras