windows下安装gpu版本tf和keras,自己安装了不少次。
1.Cuda 9.0 + cudnn 7.0
cudnn 7.0 把里面所有文件全部拉到cuda 9.0里面去,cuda会自己生成环境变量,在系统变量里面添加cuda的
lib下x64路径,但是自己要安装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