深度学习环境搭建

jzxmk / 2023-08-14 / 原文

一、Anaconda配置

镜像下载:
https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/

  • 创建虚拟环境
    创建一个名称为test的虚拟环境,如下:
conda create -n test python=3.7
  • 进入创建的虚拟环境
conda activate test
  • 离开环境
conda deactivate
  • 列举所有环境
conda env list
conda info -e
  • 删除虚拟环境
conda remove --name test --all

其中,test是要删除的虚拟环境名称,这将删除里面所有包和环境

  • 删除某个包而不是整个环境
conda remove --name <env_name> <package_name>

或者

conda uninstall <package_name>



二、深度学习框架

  • pytorch
  • tensorflow
  • 百度飞桨paddle

cuda配置

查看本地支持的cuda版本,在cmd中输入命令:

NVIDIA-smi

在anaconda虚拟环境中安装对应的cudatoolkit,一般可以向下兼容


2.1 pytorch配置

(1)pytorch gpu版安装

给当前虚拟环境中安装cudatoolkits10.2和对应的cudnn,在main和conda-forge频道中寻找要安装的软件包

conda install cudatoolkit=10.2 cudnn -c conda-forge

安装对应cuda10.2版本的pytorch

pip install torch==1.10.0 torchvision==0.11.0 torchaudio==0.10.0 --extra-index-url https://download.pytorch.org/whl/cu102

测试是否成功:
先输入python,然后键入代码

import torch
torch.cuda.is_available()

运行结果为"true"表示成功

(2)pytorch cpu版本安装

conda install pytorch torchvision torchaudio cpuonly -c pytorch

测试是否成功:
先输入python,然后键入代码

import torch
print(torch.rand(3,3))


2.2 tensorflow配置

(1)tensorflow gpu版本安装

进入虚拟环境,并输入以下指令安装python和Tensorflow-gpu版本:

conda install --channel https://mirrors.ustc.edu.cn/anaconda/pkgs/main/ python=3.7 tensorflow-gpu==2.1.0

测试是否安装成功。输入“python”后,然后键入代码

import tensorflow as tf
print("Num GPUs Available: ", len(tf.config.experimental.list_physical_devices('GPU')))

(2)tensorflow cpu版本安装

进入虚拟环境,输入以下命令安装python和tensorflow(CPU版):

conda install --channel https://mirrors.ustc.edu.cn/anaconda/pkgs/main/ python=3.7 tensorflow==2.1.0

测试是否安装成功。输入“python”后,然后键入代码

import tensorflow as tf
import os
os.environ['TF_CPP_MIN_LOG_LEVEL']='2'
print(tf.reduce_sum(tf.random.normal([1000, 1000])))

2.3 paddle配置

(1)paddle gpu版本安装

安装paddlepaddle(GPU版本),CUDA10.1以及与之配套的cuDNN

conda install paddlepaddle-gpu==2.2.2 cudatoolkit=10.1 --channel https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/Paddle/

测试是否安装成功,输入“python”后,键入代码

import paddle
paddle.utils.run_check()

(2)paddle cpu版本安装

进入虚拟环境,输入以下命令安装:

conda install paddlepaddle==2.2.2 --channel https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/Paddle/

测试是否安装成功,输入“python”后,键入代码

import paddle
paddle.utils.run_check()

参考:

  • CSDN博主「yanghanyu0126」文章 深度学习框架安装
  • 知乎用户「WalktheRainbow」文章 torch.cuda.is_available() 解决方案
  • Pytorch官网,版本对应情况 install previous versions of pytorch