默认Windows环境(Linux类似,反正Anaconda命令都一样),有支持CUDA的N卡(废话,没N卡跑什么DL www)

这篇踩坑总结只是本人在入门阶段配置环境踩过的一些坑,个人经验难免有疏漏之处,敬请海涵和斧正。

总目标:不重复踩坑,尽量追求极致便利,一步到位疾速配完环境(笑

首先在Anaconda官网:Anaconda | The World's Most Popular Data Science Platform

下载安装。

点击Win徽标在开始菜单找到Anaconda/Anaconda Prompt。

centos使用anaconda3创建了python 虚拟环境 anaconda创建pytorch虚拟环境_pytorch

conda create -n pytorch python=3.8 #创建名为pytorch的虚拟环境 python=3.8为版本限制 根据你将要安装的各种包所要求的版本而定 本文的配置是跑通过一次的
activate pytorch #激活虚拟环境 之后的操作就都在虚拟环境内进行了
conda config --remove-key channels
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 --set show_channel_urls yes
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
#以上操作均为换清华源
conda update -n base -c defaults conda #检查conda内包更新
conda install -c conda-forge mamba #安装mamba多线程下载器 加速下载
mamba install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch -c conda-forge
#11.3是目前最新版 cudatoolkit版本随Pytorch官网而定

PyTorch官网:Start Locally | PyTorch

最后一步由于不得不用官方源可能是整个过程中最慢的一步,之所以用回官方源是因为国内源大都不是最新版(表现为去掉-c后用清华源直接找不到包)。

如果显卡驱动保持较新,一般来说用最新的cudatoolkit版本就不会出CUDA版本的问题。

下载速度如果过慢建议凌晨用5G(

(2022.12更新 最近mamba下载可能会出现问题 conda也貌似更新了多线程下载功能 如果mamba下不了的话就还是用回conda install)

检验是否成功安装以及能否成功调用CUDA计算:

还是在Anaconda Prompt中:

python
import numpy
import torch
import torchvision
torch.cuda.is_available()

返回True说明安装完全成功。

如果失败,win+r打开cmd,输入:

nvidia-smi

 (Linux Shell同)

正常会打出表格

centos使用anaconda3创建了python 虚拟环境 anaconda创建pytorch虚拟环境_python_02

 对照你的CUDA版本(右上角CUDA Version)是否大于等于Pytorch官网所示建议版本(目前为11.3):

centos使用anaconda3创建了python 虚拟环境 anaconda创建pytorch虚拟环境_python_03

 如果低于该版本,建议不要纠结版本对应关系了,直接去NV官网升级显卡驱动:官方驱动 | NVIDIA

 注意这里的产品类型消费级家用卡基本都是GeForce,GeForce RTX也是GeForce(

而下面的NVIDIA RTX系列是专业卡

centos使用anaconda3创建了python 虚拟环境 anaconda创建pytorch虚拟环境_人工智能_04

顺带一提,由于torch和opencv经常联动使用,要装opencv的话直接在Anaconda Prompt中:

activate pytorch
pip install opencv-python -i https://pypi.tuna.tsinghua.edu.cn/simple/