pytorch的安装主要可以分为四步
- 安装python
- 安装Anaconda
- 安装CUDA
- 安装pytorch
1.安装python
。。。不说了,不会有任还没装python吧,不会吧。
2.安装Anaconda
https://www.anaconda.com/products/individual点击链接进入下载地址
根据自己的系统进行选择,选好后等着下完,可能会有点慢。
打开后一直下一步
路径最好放在python下
最后安装完毕就行(抱歉安装成功的图我忘记截了)
安装CUDA
下载地址https://developer.nvidia.com/cuda-toolkit-archive
进入下载页面选择对应的版本号
本来是等他直接下载完就好了,但是点了download之后半天没反应,我在网上找了一下发现大家都下不了,不知道是英伟达在搞事情还是出现了bug。
解决方法一
点了几个链接试了一下,CUDA Toolkit 9.0可以用,所以我就下下来了
解决方法二
在网上找大佬下好的发过来,或者是自己私下找已经下过的人拷一份
反正下载好了以后直接安装就完事,安装完成后添加CUDA的系统变量
在此电脑上点击右键→属性,进入”系统“
点击高级系统设置
环境变量
把这玩意配置到环境变量中
键入%CUDA_PATH%\bin\cudart64_90.dll,编辑完成后如图所示
进入cmd窗口,键入nvcc -V
版本号 V9.0.176,表示安装成功
安装好Anaconda和CUDA之后,就可以进入pytorch的安装了
安装pytorch
进入官网选择版本会看到这样一个界面
根据自己的CUDA版本选择对应的命令,这里我用的如上图所示
如何查看自己的CUDA版本见另一篇文章
https://blog.csdn.net/bruce_zhao1407/article/details/109580835
打开cmd运行
然后在dos窗口输入python,再输入以下命令检测一下是否安装成功
看到最底下的true就完事啦~~
注:以上只是一个基本的流程,我安装过程中也出现了很多问题,通过查阅自来哦还是都解决了,但是没有办法一一打出来了,欢迎大家一起讨论