(一)开发环境
Python3.7+,Cuda10.0+,Pycharm
(二)安装
1、Anaconda5.3.1+
https://www.anaconda.com/products/individual
2、Cuda10.0+
https://developer.nvidia.com/cuda-downloads
确认文件bin\nvcc.exe存在于安装目录中;
添加文件夹bin、libnvpp到系统路径;
命令行输入nvcc -V。
3、Pytroch
https://pytorch.org/get-started/locally/
· 国外用户
conda install pytorch torchvision cudatoolkit=10.2 -c pytorch
· 国内用户
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes
conda install pytorch torchvision cudatoolkit=10.2