上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,安装完成。