当前位置： 移动技术网 > IT编程>网页制作>HTML > PyTorch环境搭建及踩过的坑

PyTorch环境搭建及踩过的坑

2020年07月18日  | 移动技术网IT编程  | 我要评论

以下过程仅限windows

1、安装

总体上的PyTorch安装我是参考的这篇博客：https://www.cnblogs.com/zhouzhiyao/p/11784055.html

【踩坑一】官网上没有和我的cuda匹配的版本。
网上搜了一下也没有解决这个问题，索性就用10.2吧，以后搞清楚了在这里把坑补上。
在这里插入图片描述
在这里插入图片描述

【踩坑二】pytorch慢到无法安装
这里我参考了 https://blog.csdn.net/xo3ylAF9kGs/article/details/104104041/
注意点1：这几个镜像源的地址，贴一下：

channels:
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/

注意点2：

已经都安装好镜像源，但是为什么安装还是龟速？ 原因是 conda install pytorch torchvision
cudatoolkit=10.1 -c pytorch 命令中，最后是-c
pytorch，所以默认还是从conda源下载，新安装的清华等源没有用上。

正确命令：conda install pytorch torchvision cudatoolkit=10.1，也就是去掉-c
pytorch

并且在安装时，也能看到使用了清华源。并且安装速度直线提升，顺利done

注意点3：
如果像下图这样安不上，那就多重复执行几次命令，总会安上的。
在这里插入图片描述

2、虚拟环境里怎么用jupyter notebook？

好，终于安装成功了。但我把PyTorch安在了虚拟环境里了啊，没法用外面的jupyter notebook了，咋办？
网上提供的方法还是挺多的，我主要参考了这几篇文章：
https://www.pianshen.com/article/3741314372/
https://www.cnblogs.com/yzl0/p/11490752.html
https://www.icode9.com/content-3-71611.html

经过我的多次对比，有管用的方法也有不管用的方法，下面总结了一个最简单、最适合我的方法：

在你的虚拟环境里（注意，都是在虚拟环境里操作）输入两条命令：

conda install ipykernel

python -m ipykernel install --name pytorchEnv

那个 pytorchEnv 是此虚拟环境显示在Jyputer中的名称，可自定义

即可。
然后打开jupyter notebook，右上角创建或者进入后切换内核都是可以的。
在这里插入图片描述

【删除内核】比如那个test虚拟环境我不想要了，删掉了，但是jupyternotebook里还显示test这个内核（其实已经不管用了），如何从其列表中删除掉呢？
打开一个命令行窗口，输入

jupyter kernelspec remove test

即可删除。
在这里插入图片描述

现在可以查看一下内核

jupyter kernelspec list

本文地址：https://blog.csdn.net/qq_38237214/article/details/107371113

您可能感兴趣的文章:

如对本文有疑问， 点击进行留言回复！！

相关文章:

验证码：
最近更新的文章
大家感兴趣的文章
移动技术网