ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

绕过sudo安装pytorch(GPU)

2022-02-28 11:05:46  阅读:225  来源: 互联网

标签:torch 传送门 sudo pytorch cuda 版本 GPU True


最近几天走了n多弯路 大概做一下总结

1.如果是n卡的话就nvidia-smi 看一下driver版本
根据driver版本再选择cuda版本
在这里插入图片描述

(学校的没有公用cuda 装了几次新版才发现这个问题
具体版本的下载地址百度/Google一下都能找到
因为没有sudo 所以要把路径都放到自己的文件夹里 也不要尝试写入自己以外的空间
用nvcc -V检测安装情况

2.装好了cuda之后对照着自己的情况找对应的包
传送门1
有时候下载太慢或者没有对应的版本 可以自己下载了再传过去 pip install xxx.whl
传送门2
一般要装 torch和torchvision 手动找的时候要注意cp python版本和cu cuda版本

import torch
print(torch.cuda.is_available())
#True

返回True就差不多可以告一段落了~

标签:torch,传送门,sudo,pytorch,cuda,版本,GPU,True
来源: https://blog.csdn.net/weixin_45817880/article/details/123177324

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有