pytorch的开发环境安装

一、Anaconda+python3.7

https://www.anaconda.com,下载Anaconda，安装时，注意勾选加入运行环境。

安装后，在cmd中，输入conda，命令成功说明安装成功。

二、安装cuda.10.1（显卡使用gtx-1050ti，入门）

https://developer.nvidia.com/cuda-downloads?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal

下载运行即可，安装好后，在运行环境的path中确认是否有安装好的路径。

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\bin，没有的话完成添加。

在cmd中输nvcc -v 确认安装是否成功，成功的话会输出版本号。

 

三、安装pytorch

https://pytorch.org，

cmd右键，管理员权限运行，输入：

conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

完成安装。若安装失败，可以用pip的方式。

 

四、安装pycharm

https://www.jetbrains.com/pycharm/官网下载pycharm的社区版，安装好后；

选择exsiting interpreter->

选择conda envirnment-〉 interpreter-〉

路径：手动输入conda的安装路径

找到python.exe，确定。

 

最后，在pycharm新建一个工程，新建main.py，输入import pytorch，成功即为安装成功。