一、在服务器上安装Anaconda3
1、使用wget 下载链接命令下载Anaconda的安装包
通过复制Anaconda官网下载Anaconda:
wget https://repo.anaconda.com/archive/Anaconda3-2020.07-Linux-x86_64.sh
或通过清华源上下载:
wget https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/Anaconda3-2020.07-Linux-x86_64.sh
2、为下载的文件添加可执行权限
chmod u+x Anaconda3-2020.07-Linux-x86_64.sh
3、开始安装Anaconda
./Anaconda3-2020.07-Linux-x86_64.sh
4、指定 Anaconda 的安装路径
/home/Anaconda3
回车后提示是否接受许可文件,输入yes回车
这里是协议内容,一直回车到底部,然后输入yes —> 回车
选择安装路径,在下列输入要安装的路径
安装完成
5、修改.bashrc文件,将conda路径添加到环境变量中(因第四步中加入了环境变量故可省略)
先打开home下的.bashrc文件
vi /home/.bashrc
输入i,然后在最后输入
export PATH="/home/anaconda3/bin$PATH:"
使用Esc退出编辑,然后输入“:wq”保存并退出.bashrc的编辑
6、检查环境变量是否配好
source ~/.bashrc
conda info --envs
二、在Anaconda3上创建虚拟环境
1、创建虚拟环境
conda create -n tf37(环境名称) python=3.7(版本号)
输入y即可
查看创建的虚拟环境:
conda info --envs
2、激活虚拟环境
conda activate tf37
3、安装cuda和cudnn(cudatoolkit→cudnn→tensorflow-gpu)
安装cudatoolkit的代码:
conda install cudatoolkit=10.1
安装cudnn的代码:
conda install cudnn=7.6.5
4、下载tensorflow-gpu:
pip install tensorflow-gpu==2.3.0 -i https://pypi.douban.com/simple
或在官网下载离线文件,然后进入到存放.whl文件下,然后输入如下命令进行安装
pip install tensorflow_gpu-2.4.0-cp38-cp38-manylinux2010_x86_64.whl
5、测试是否安装成功
输入python,然后一条一条输入
import tensorflow as tf
tf.__version__ #测试cpu版
tf.test.is_gpu_available() #测试GPU
安装成功。
三、安装pytorch-gpu
pytorch官网:https://pytorch.org/
1、安装pytorch
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
2、测试是否安装成功
import torch
print(torch.__version__)
print(torch.cuda.is_available())