本人也是第一次搭建环境,遇到很多问题,分享给大家,希望大家少走弯路,请在此之前安装好conda,官网下载一步步安装即可,conda官网Anaconda | Anaconda Distribution
有显卡驱动的直接打开控制面板,可以看到自己显卡的配置信息例如我这里是gtx 1650
(一定要根据自己显卡下载对应的版本)
没有显卡驱动的
电脑右击-->管理-->设备管理器-->显示适配器。就可以看到电脑显卡的版本
没有驱动的到官方下载驱动
官方驱动 | NVIDIA
根据自己显卡配置选择,notebooks是笔记本的意思,用笔记本的选
然后一步步下一步安装
安装(更新)好了显卡驱动以后。我们按下win+R组合键,打开cmd命令窗口。输入如下的命令
查看版本信息
nvidia-smi
驱动的版本是452.30;最高支持的CUDA版本是11.2版本。得到显卡的最高支持的CUDA版本为11.2
在菜单里,打开anaconda prompt
查看自己的环境
conda env list
这里我已经搭建好嘞pytorch和paddle环境所以会显示
创建虚拟环境conda create -n 环境名字(英文) python=x.x(python版本),我就是创建了一个名字叫pytorch的环境,python是3.8版本的环境。
conda create -n pytorch python=3.8
会有很多基础的包,询问是否安装的时候,输入y。就可以创建环境了这个环境了
激活这个环境,输入命令conda activate pytorch
conda activate pytorch
可以看到我们进入了刚刚创建的pytorch环境
pytorch的官网在国外,下载相关的环境包是比较慢的,所以我们给环境换源。在pytorch环境下执行如下的命名给环境换清华源中科源都可以
我这里用的清华源
在pytorch环境输入以下代码
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes
pytorch的官网PyTorch,前面已经查询过我的,最高支持cuda11.2版本,所以我选择cuda11.1
在pytorch环境输入以下代码
pip install torch==1.10.0+cu111 torchvision==0.11.0+cu111 torchaudio==0.10.0 -f https://download.pytorch.org/whl/torch_stable.html
我这里用conda install时候有报错,conda install 包 改用>>pip install 包
检验是否安装成功,在pytorch中运行python
运行如下代码,查看是否anconda在安装pytorch环境的时候也安装了cuda和cudnn。
import torch
print(torch.cuda.is_available())
print(torch.backends.cudnn.is_available())
print(torch.cuda_version)
print(torch.backends.cudnn.version())
可以看到cuda和cudnn已经安装,并且可以得到cuda的版本为11.1和cudnn的版本为8.05版本
和pytorch,一样,在base环境下创建一个新的环境来安装paddlepaddle框架
conda create -n paddle python=3.8
conda activate paddle
进入飞桨PaddlePaddle-源于产业实践的开源深度学习平台网站
我这里选的cuda 11.1 ,复制图上的命令
在paddle环境里面输入复制的命令
python -m pip install paddlepaddle-gpu==2.3.2.post111 -f https://www.paddlepaddle.org.cn/whl/windows/mkl/avx/stable.html
安装即可,到这里两个环境已搭建完成