AI源码解读:数字图像处理案例(Python版)
上QQ阅读APP看书,第一时间看更新
 AI源码解读.数字图像处理案例:Python版

1.2.2 PyTorch环境

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

创建Python 3.7的环境,名称为PyTorch:

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

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

     activate pytorch

安装PyTorch和cuda 10.1,输入命令:

     conda installtorch torchvision cudatoolkit=10.1

安装完毕。