win10 安装GPU版pytorch踩坑

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/jining11/article/details/88125541

其实我现在还是有些忐忑的,毕竟我的英伟达老显卡算力只有5.0,希望我的电脑不会跑着跑着就散出一股诱人的焦香叭。

Step1: 安装清华大学镜像Anoconda

地址:https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/
(注意最新的在最下面)

Step2:运行Anaconda Prompt并搭建虚拟环境

在这里插入图片描述

一路yes

在这里插入图片描述

Step3:下载网盘的pytorch源

地址:https://pan.baidu.com/s/1dF6ayLr#list/path=%2F&parentPath=%2F

里面的东西不需要全下,需要什么版本就下什么版本,首先要看一下自己的显卡够不够的上5.0,以决定是CPU版还是GPU版,如果是GPU版,再看看是cu80还是cu90,然后选择python3.6对应的版本下载到本地,比如我的就是pytorch-0.3.0-py36_0.3.0cu80.tar.bz2

Step4: 开启虚拟环境并进入源所在文件夹(我的在train/pytorch里面)

  • activate torch
  • cd XXX
  • conda install numpy mkl cffi
    在这里插入图片描述

离线安装及测试

  • 安装
    在这里插入图片描述
  • 测试
    在这里插入图片描述

猜你喜欢

转载自blog.csdn.net/jining11/article/details/88125541