Pytorch(一) Pytorch 的安装

奋斗的周周 2020-01-13 23:57:19 32668 收藏 47
分类专栏： 机器学习/深度学习
版权
安装步骤
1.检查是否有合适的GPU, 若有安装Cuda与CuDNN

2.下载pytorch安装文件，进行pytorch的安装

检查电脑是否有合适的GPU
在桌面上右击如果能找到NVIDA控制面板，则说明该电脑有GPU。控制面板如下，并通过查看系统信息获取支持的Cuda版本

 

点击 帮助->点击 系统信息 弹出下面的对话框，在驱动程序版本那一栏就能看到该计算机使用的驱动版本。

 

 

下载Cuda
官网：https://developer.nvidia.com/cuda-10.1-download-archive-update2

在https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html 这里可以查询到我们应该下载哪个版本

 

然后，根据实际情况选择合适的版本

 

下载CuDNN 
官网 https://developer.nvidia.com/rdp/cudnn-download

 

安装Cuda
1.与安装其他的软件类似

2.安装结束后将  ~/nvcc/bin(因为版本的不同可能在不同的地方) 目录添加到环境变量

3.在命令行下输入 nvcc -V, 出现下列信息说明Cuda安装成功

 

4.将CuDNN压缩包解压后，下面的三个文件夹复制到Cuda的安装目录下

 

5.然后执行Demo, 如果Demo中显示PASS则说明安装成功

 

 

安装Pytorch
官网 https://download.pytorch.org/whl/torch_stable.html 选择合适的版本

torch/torchvision 都需要安装

 

 

 

 

验证Pytorch

 

安装成功
通过上面输出的信息，说明安装是成功的
————————————————
版权声明：本文为CSDN博主「奋斗的周周」的原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接及本声明。
原文链接：https://blog.csdn.net/qq_23013309/java/article/details/103965619

版权声明：本文为think90原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://www.cnblogs.com/think90/articles/13223377.html