1、安装步骤
(1) 打开Anaconda Prompt
在开始菜单中点击打开Anaconda3(64-bit)下的 Anaconda Prompt(Anaconda) 命令行工具:
(2) 安装Pytorch
打开Pytorch官网,点击Get Started,选择所需版本的conda下载命令,如下图所示:
使用Ctrl+C复制此命令,然后使用Ctrl+V将其粘贴到Anaconda Prompt(Anaconda) 命令行中:
(base) C:\Users\86151>conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
等待安装完成即可。
2、验证是否安装成功
方法1
打开Anaconda客户端,点击Enviroments,搜索torch,直接查看即可:
方法2
在 Anaconda Prompt(Anaconda) 命令行中输入:
(base) C:\Users\86151>conda list
可以看到已经成功安装pytorch: