此安装说明适用于刚接触pytorch,python的朋友们:
安装pytorch前需要以下几个提前准备步骤
1.安装python3.6 进入官网下载版本https://www.python.org/downloads/windows/ 可根据自己的需要下载不同的版本
2.安装anaconda对应的python3.6版本
链接: https://pan.baidu.com/s/1O3Zwbb0hFDJ0K_wAPDudlw 密码: RSXS
此网盘连接对应着Anaconda5.2.0-windows-x86.exe
具体anaconda安装步骤可参考https://blog.csdn.net/qq_42543252/article/details/80806536
3.如果您的电脑装有NVDIA显卡,可安装CUDA进行GPU加速
相应的CUDA9.2下载连接为:https://developer.nvidia.com/cuda-downloads 可根据自己的电脑配置选择自己的CUDA版本
4.安装pytorch在windows环境下
下载连接为:https://pytorch.org/ 可根据自己的配置选择,如您是按照以上步骤进行安装的可以选择如下配置:
打开之前安装好的anaconda中的anaconda prompt,如下图所示:
进入到你想安装的文件夹当中(例如安装到D盘下)
输入:D:
输入:conda install pytorch cuda92 -c pytorch
输入:pip3 install torchvision
注意:博主在安装时出现如下BUG
如出现这种错误可输入:python -m pip install --upgrade pip 对pip进行更新
而后方可安装成功!
感谢文中引用链接博主的贡献!希望可以帮助大家!