1 准备工作
在安装Pytorch前,首先准备安装好anaconda。步骤较为简单,不熟悉的可以直接参照这篇文章《Anaconda介绍、安装及使用教程》进行安装。这篇文章非常详细,mac和windows的安装、管理及卸载里面都说明清楚了,完全可以自行解决。
PS:国内安装anaconda建议使用**清华镜像**。
2 安装Pycharm
官网下载安装包 官网下载安装包 官网下载安装包 官网下载安装包 官网下载安装包 官网下载安装包 官网下载安装包 https://www.jetbrains.com/pycharm/安装即可。
3 安装PyTorch
在1.2版本以后,PyTorch只支持cuda 9.2以上了,所以需要对cuda进行升级,目前测试大部分显卡都可以用。
3.1 安装步骤:
- 检查是否有核实的GPU,如果有则需要安装CUDA与CUDNN,如果没有则不需要安装
- 选择合适的插件下载并通过pip或conda进行安装。
#默认 使用 cuda10.1
pip3 install torch=1.3.0 torchvision=0.