主要是参考了这篇文章win10+pytorch CUDA
期间,安装CUDA时目录选择出错了导致安装在了C盘的temp目录下
之后出了一个问题,
![10651191-0b521b1928dcc359.png](https://i-blog.csdnimg.cn/blog_migrate/3b73e9e8da78d4194e544b523ba25ea3.png)
环境变量
安装程序提示我这个环境变量太长了它加不进去,所以手动尝试加了一下环境变量
添加环境变量时遇到了环境变量的字符过长的问题,参考了解决
添加完环境变量后终于可以正确显示CUDA的版本了
zz :: ~ » nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2019 NVIDIA Corporation
Built on Sun_Jul_28_19:12:52_Pacific_Daylight_Time_2019
Cuda compilation tools, release 10.1, V10.1.243
之后想要进一步安装Cudnn,因为优化做的更好,但是感觉太茫然就没有尝试了
去Pytorch官网选择一下就可以安装了
![10651191-b2c7c1ba5b7d70f3.png](https://i-blog.csdnimg.cn/blog_migrate/105ec32d6f930ab28d2706c6d38685ec.png)
pytorch官网
之前因为安装了一个torch,但是是不支持cuda的这次重新装一个(不需要手动卸载,pip会自动检测到然后卸载掉老的版本
![10651191-e285df64c95877a1.png](https://i-blog.csdnimg.cn/blog_migrate/8425c92ebcb8b3bf13eb118184415d00.png)
torch测试能够使用cuda
安装时候遇到了这个错误
![10651191-4e0730d090b39745.png](https://i-blog.csdnimg.cn/blog_migrate/01f249c51b51a53a93f7c7256a1bcb30.png)
error
zz :: ~ » python
Python 3.5.4 (v3.5.4:3f56838, Aug 8 2017, 02:17:05) [MSC v.1900 64 bit (AMD64)] on win32
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> torch.__version__
'1.2.0'
>>> torch.cuda.is_available()
True
终于安装成功了