Windows下使用conda在虚拟环境中安装CUDA、CUDNN及Tensorflow

喜大普奔!!! 使用anaconda直接安装深度学习环境所需的cuda、cudnn、tensorflow-gpu,windows,ubuntu都可以。参考文章

需要anaconda,以及nvidia的驱动。如果你是ubuntu的话,可以参考这个文章装nvidia驱动。

1.创建环境

conda create -n your_env_name python=X.X
如:
conda create -n starcraft python=3.5

2.激活环境:

activate your_env_name
如:
activate starcraft

3.安装cuda和cudnn

conda install cudatoolkit=9.0 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/linux-64/
conda install cudnn=7.1.4 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/linux-64/

4.安装tensorflow-gpu(这里需要和之前的版本对应到)

pip install tensorflow-gpu==1.9.0

 

 

 

猜你喜欢

转载自blog.csdn.net/weixin_39059031/article/details/86565922