windows下安装gpu版本tf和keras,自己安装了不少次。

1.Cuda 9.0 + cudnn 7.0

cudnn 7.0 把里面所有文件全部拉到cuda 9.0里面去,cuda会自己生成环境变量,在系统变量里面添加cuda

libx64路径,但是自己要安装vs,我安装了vs2017,在cuda.exe安装中应该会自动编译。

2. Anaconda

3. Anaconda里  pip install tensorflow_gpu == 1.11.0,pip install keras(conda install tensorflow_gpu)

4. Anaconda里  pytorch conda install pytorch cuda90 –c pytorch

 

Logo

腾讯云面向开发者汇聚海量精品云计算使用和开发经验,营造开放的云计算技术生态圈。

更多推荐