Windows安装pytorch

  新手第一次安装,记录自己安装经过。

  1.安装anaconda

   直接在anaconda官网上选择相应版本安装包安装即可。anaconda自带python,为避免和已有python冲突,可将已有版本删除。anaconda安装时记得一定要勾选上添加环境变量这一选项。成功安装后可在命令行中输入conda --version进行测试。

  2.安装CUDA和cudnn

  CUDA我之前装过了,直接就跳过了。不过安装时注意选择自定义安装,之后一路下一步就好。安装后的位置如图。

  

  cudnn去官网下载,下载时要完成一个简单的注册才能下载。注意下载时选择和自己CUDA匹配的版本。下载下来是一个压缩包,只需要把每个文件夹的内容复制到CUDA中同名文件夹下即可完成安装,cudnn中单独的那个文件复制到上图文件夹中即可。

  3.创建虚拟环境

  打开anaconda prompt或者命令行,输入

1
conda create -n pytorchlearn(虚拟环境名) python=3.9

  完成创建虚拟环境,之后输入

1
activate pytorchlearn

  即可激活环境。

  4.安装torch和torchvision

  在命令提示符中先pip安装wheel包,之后直接pip先安装torch,再安装torchvision。可能是使用了梯子的原因,安装速度不慢。

  完成安装。

  

 

 

  

 

作者:y0hv2y

出处:https://www.cnblogs.com/y0hv2y/p/17028958.html

版权:本作品采用「署名-非商业性使用-相同方式共享 4.0 国际」许可协议进行许可。

posted @   yuhury  阅读(57)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· 阿里最新开源QwQ-32B,效果媲美deepseek-r1满血版,部署成本又又又降低了!
· 开源Multi-agent AI智能体框架aevatar.ai,欢迎大家贡献代码
· Manus重磅发布:全球首款通用AI代理技术深度解析与实战指南
· 被坑几百块钱后,我竟然真的恢复了删除的微信聊天记录!
· AI技术革命,工作效率10个最佳AI工具
more_horiz
keyboard_arrow_up light_mode palette
选择主题
点击右上角即可分享
微信分享提示