先附两个api
pytorch API
pytorch api
开发环境安装(windows)
- 安装anaconda
- 勾选 Add Anaconda to my PATH environment variable
- 安装CUDA10
- 到CUDA官网下载,选择local 版本
- 安装过程中如果没有CUDA Visual Studio Integration 可以勾选然后跳过
- 如果在cuda>v10.0>bin 目录下找到nvcc.exe 则安装成功。也可以在命令提示符中输入nvcc -V 命令,看是否有反应来查看是否安装成功。
- 安装pytorch
- 到pytorch官网,按提示选择对应版本,复制命令
- 在命令行输入命令,进行安装。如果安装不成功可以使用镜像来安装(注意:安装时去掉 -c pytorch 字符才能使用镜像来安装,不然还会走官网)。
- 安装pycharm
- 到pycharm 官网,下载安装,双击安装包按提示安装即可。