PyTorch安装教程(最简单方法)
2023-04-18 15:43:41 时间
注意:在安装之前确定你要安装的pytorch版本,然后在PyTorch和CUDA版本对应关系中查找对应版本,再进入正式安装环节
之前使用anaconda进行安装,下载速度实在太慢,安装几次都失败了。后听我朋友建议采用pip进行安装,直接成功!!!
一、根据之前你查看的对应版本,安装对应的 cuda 和 cudnn(我当时安装是参考的1-cuda和cudnn安装和2-cuda和cudnn安装)
二、cuda 和 cudnn都安装成功后,采用pip安装PyTorch
1、查看 cudnn 和 cuda 版本
#查看cudnn 版本
nvcc -V
#查看cuda版本
set cuda
2、进入pytorch官网:https://pytorch.org/,下载对应版本(PyTorch和CUDA版本对应关系)
- 网页下拉,即可看到下图,可安装的CUDA版本,给出了安装命令。如果没有找到对应版本,或者该版本没有安装命令,则进行下一步
- 查看所有版本
跳转进入下面页面
下滑找到需要的pytorch版本
继续下滑,使用pip安装,我的是win11系统+cuda10.2+pytorch1.9.0,复制对应的安装代码
2、打开PyCharm软件,打开终端,粘贴安装代码
等待安装,我当时安装大概十分钟左右(用conda安装好几个小时还安装失败!!)
3、查看安装是否成功:
- 使用
pip list
检查
- 不确定的话,还可以进行代码测试:
import torch
print(torch.__version__)
print(torch.cuda.is_available())
print(torch.version.cuda)
这样就成功啦!!
这是小新学习的小记录,有不对的地方希望UU们批评指正
相关文章
- 【技术种草】cdn+轻量服务器+hugo=让博客“云原生”一下
- CLB运维&运营最佳实践 ---访问日志大洞察
- vnc方式登陆服务器
- 轻松学排序算法:眼睛直观感受几种常用排序算法
- 十二个经典的大数据项目
- 为什么使用 CDN 内容分发网络?
- 大数据——大数据默认端口号列表
- Weld 1.1.5.Final,JSR-299 的框架
- JavaFX 2012:彻底开源
- 提升as3程序性能的十大要点
- 通过凸面几何学进行独立于边际的在线多类学习
- 利用行动影响的规律性和部分已知的模型进行离线强化学习
- ModelLight:基于模型的交通信号控制的元强化学习
- 浅谈Visual Source Safe项目分支
- 基于先验知识的递归卡尔曼滤波的代理人联合状态和输入估计
- 结合网络结构和非线性恢复来提高声誉评估的性能
- 最佳实践丨云开发CloudBase多环境管理实践
- TimeVAE:用于生成多变量时间序列的变异自动编码器
- 具有线性阈值激活的神经网络:结构和算法
- 内网渗透之横向移动 -- 从域外向域内进行密码喷洒攻击