标签:python torch CUDA 版本 pip 显卡 官网 安装
一、 安装方法
1、首先进入torch官网,选择合适的版本,由于我是Windows系统,想跑跑GPU,就选择Windows和CUDA的选项,复制下图红色方框处的命令。
PS:关于CUDA版本的选择,可以参考Pytorch版本、CUDA版本与显卡驱动版本的对应关系,若不知道自己显卡驱动版本,可以参考怎样看自己电脑显卡驱动程序的型号和版本??,如果发现自己显卡驱动程序的表示方式不一样,可以上Geforce官网根据自己显卡的型号,搜索对应的显卡驱动并下载。
2、 win+R,输入cmd(也可以在电脑左下方直接搜索cmd),将命令粘贴直接回车即可安装。
二、 报错及解决
若遇到无法连接网站".........ERROR: Could not find a version that satisfies the requirement torch",或者下载很慢下一半断开连接,出现很多exception。
可以打开官网,ctrl+f搜索并下载版本对应的.whl文件(torch/torchvision/torchaudio的版本号在图1红色方框中的命令行中都有),下载完成之后,Win+R 输入"pip install 文件名称.whl",与上面的安装方法类似,即可成功安装!!!
三、模块测试
最后可以用下面的代码测试是否安装成功:
import torch
print(torch.__version__) # 这个输出torch版本
print(torch.cuda.is_available()) # 这个输出结果必须为True
希望我的分享能够帮到您,如果觉得写的不错可以多多点赞支持,谢谢您的观看!!!
标签:python,torch,CUDA,版本,pip,显卡,官网,安装 来源: https://blog.csdn.net/weixin_44178960/article/details/117353665
本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享; 2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关; 3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关; 4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除; 5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。