windows安装CUDA

主要是参考了这篇文章win10+pytorch CUDA

期间,安装CUDA时目录选择出错了导致安装在了C盘的temp目录下

之后出了一个问题,


10651191-0b521b1928dcc359.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
pytorch官网

之前因为安装了一个torch,但是是不支持cuda的这次重新装一个(不需要手动卸载,pip会自动检测到然后卸载掉老的版本

10651191-e285df64c95877a1.png
torch测试能够使用cuda

安装时候遇到了这个错误

10651191-4e0730d090b39745.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

终于安装成功了

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值