AI源码解读:卷积神经网络(CNN)深度学习案例:Python版
上QQ阅读APP看书,第一时间看更新

2.2.2 PyTorch环境

打开Anaconda Prompt,添加清华仓库镜像,输入命令:

    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

创建新的Python环境,名称为PyTorch,Python版本号要对应,输入命令:

    conda create-n pytorch python=3.6

有需要确认的地方,都输入y。

在Anaconda Prompt中激活PyTorch环境,输入命令:

    activate pytorch

在PyTorch官网查看可供选择PyTorch版本、系统版本、安装方式以及编程语言,获得下载命令,还可选择CUDA加速,下载地址为:https://pytorch.org/get-started/locally/。

选用Stable 1.4版本、Windows系统、conda安装包、Python语言以及CUDA 10.1版本的PyTorch,在Anaconda Prompt中安装,输入命令:

    conda install pytorch torchvision cudatoolkit=10.1

有需要确认的地方,都输入y,安装完成。