【动手学AI】安装CUDA 和pytorch环境

欢迎使用

你好！ 先安装CUDA11.6版本，再安装Anaconda，再安装pytorch

conda数据源管理
下面展示一些 内联代码片。

#显示目前conda的数据源有哪些
conda config --show channels
#添加数据源：例如, 添加清华anaconda镜像：
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 --remove channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/

`` 安装pytorch
conda install pytorch torchvision torchaudio cudatoolkit=11.6

验证是否安装好

import torch
torch.cuda.is_available()
#cuda是否可用；
torch.cuda.device_count()
#返回gpu数量；
torch.cuda.get_device_name(0)
#返回gpu名字，设备索引默认从0开始；
torch.cuda.current_device()
#返回当前设备索引；