从0安装pytorch
1. 检查自己的电脑有没有GPU
- 首先打开任务管理器,选择性能栏
- 然后滑到最下,看是否有GPU一项(我的电脑是AMD的核显,用来入门深度学习)
2. 安装CPU版的pytorch
然后我发现,好像AMD的核显没办法用CUDA(这是英伟达为他们的芯片设计的计算平台),所以好像这个核显=没有显卡…
于是我们开始来安装CPU版的pytorch(用于入门)
首先打开pytorch官网,点击install
然后,选择对应的环境
最后,将生成的命令复制到cmd中运行
3. 测试pytorch
我们测试torch中最常用的张量
观察到
说明安装成功