(base) will@ubuntu-hp:~$ python
Python 3.8.8 (default, Apr 13 2021, 19:58:26)
[GCC 7.3.0] :: Anaconda, Inc. on linux
Type “help”, “copyright”, “credits” or “license” for more information.
- 创建新的conda环境,名为py38:
conda create -n py38 python=3.8.8
- 激活py38:
conda activate py38
- 安装指定版本的tensorflow,指定国内源以加快下载速度:
pip install tensorflow-gpu==2.3.0 -i https://pypi.tuna.tsinghua.edu.cn/simple
- 安装CUDA:
conda install cudatoolkit=10.1 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/linux-64/
- 安装cudnn:
conda install cudnn=7.6.5
- 安装完成,接下来验证一下是否GPU版的TensorFlow是否安装成功
验证
-
退出ssh,重新登录
-
查看有哪些conda环境,以及正在使用哪个,输入命令conda info --e,如下