win10,在anaconda虚拟环境中安装特定的深度学习版本(以TensorFlow为例)

以下内容是在已经安装好anaconda软件后配置的

一、安装cuda及cudnn

1.离线安装

1.根据项目要求,先利用官网安装好需要的cuda版本,不需要配置环境变量

2.经过实践证明,通过conda命令安装的cuda,不能完成配置,cuda仍为系统环境变量配置版本

2.anaconda虚拟环境中

a.创建虚拟环境:打开anaconda prompt,输入:
 conda create -n tensorflow python=3.7 # tensorflow为虚拟环境名称可自定义;python版本也可自己设定
b.激活虚拟环境并安装cuda和cudnn
 activate tensorflow  #激活虚拟环境
 conda install cudatoolkit=10.1 # 安装需要的cuda对应版本,与之前离线安装版本对应
 conda install cudnn=7.6.0 # 安装cudnn 对应版本可网上查阅
c.cuda版本也可自行选择,cudnn和cuda版本对应请自行百度。如果下载很慢容易断开,建议搭个梯子或者使用清华镜像源:
 conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
 conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
 conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
d.检测cuda是否安装成功:
 conda list

注:

1.经实践证明,可以直接安装tensorflow-gpu,即直接执行步骤二,其对应的cuda与cudnn可自动配置安装。

2.若配置其他框架,仍需按顺序安装。

 

二、安装tensorflow-gpu

1.tensorflow与cuda、cudnn、pyhton的版本都有对应关系,具体对应关系可上网查阅
 conda install tensorflow-gpu==2.3.0
2.numpy安装

tensorflow与numpy有版本对应关系,可以先安装tensorflow后,运行程序报错再按需求选择较低版本安装;或者直接在安装tensorflow前安装合适版本的numpy直接安装。

 pip install numpy==1.16.0 # 安装numpy版本,安装之前请先卸载
 pip uninstall numpy # 卸载numpy
 conda list # 显示新建虚拟环境所有已安装模块
3.验证
 import tensorflow as tf
 version = tf.__version__
 gpu_ok = tf.test.is_gpu_available()
 print("tf version:",version,"\nuse GPU",gpu_ok)

三、为不同虚拟环境配置不同的cuda

1.打开anaconda prompt,进入需要配置的虚拟环境:
 activate tensorflow  #激活虚拟环境
2.将prompt窗口定位到虚拟环镜主目录下:
 cd Anaconda3\envs\tensorflow #anaconda默认安装在c盘
 cd /d D:\Users\Anaconda3\envs\tensorflow #anaconda安装在其他位置
 显示效果:(tensorflow) D:\Users\Anaconda3\envs\tensorflow>
3.执行以下命令:
 mkdir .\etc\conda\activate.d #创建activate.d文件夹
 mkdir .\etc\conda\deactivate.d #创建deactivate.d文件夹
4.在activate.d文件夹中创建 env_vars.bat文件,将cuda版本添加到临时系统变量中。

a.创建.bat文件方法:右键->新建文本文档->另存为->文件名.bat,保存类型:所有文件

b.内容:

 @set CUDA_PATH=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\bin
 @set CUDA_NVVP=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\libnvvp
 @set CUDA_lib=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\lib\x64
 @set OLD_PATH=%PATH%
 @set PATH=%CUDA_PATH%;%CUDA_NVVP%;%CUDA_lib%;%PATH%;
5.在deactivate.d文件夹中创建同名文件env_vars.bat,将环境变量重置为原始的内容。

文件内容:

 @set PATH=%OLD_PATH%
6.测试

分别在base环境与新建的tensorflow环境输入:

 nvcc --version

若base环境显示的cuda版本为之前添加环境变量的版本,而新虚拟环境的版本为上述步骤新建的,则配置成功。

四、如何在pycharm中选定已配置的虚拟环境

1.新建的虚拟环境默认保存位置在anaconda安装路径中的envs文件

2.打开pycharm后按照以下路径操作:

a.file->setting->project:新建项目名->python interpreter->add

b.Virtualenv Environment ->existing environment->新建虚拟环境路径 ->.....python.exe ->ok

 
posted @ 2022-05-31 10:05  Whp_bicycle  阅读(135)  评论(0编辑  收藏  举报