Loading

Windows安装pytorch

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

  1.安装anaconda

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

  2.安装CUDA和cudnn

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

  

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

  3.创建虚拟环境

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

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

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

activate pytorchlearn

  即可激活环境。

  4.安装torch和torchvision

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

  完成安装。

  

 

 

  

 

posted @ 2023-01-05 22:16  yuhury  阅读(46)  评论(0编辑  收藏  举报