1 深度学习的工作环境,一般流程

  • 安装好conda
  • 创建虚拟环境并且**
    conda create -n pytorch python=3.6
    source activate pytorch

.
需要知道,使用conda命令安装库时,会安装其他的依赖库(或相关库),而且这些库也有默认的版本。所以在多个库相互兼容的情况下,想要安装指定版本的库,可以使用pip进行单独安装

2 安装pytorch

进入pytorch官网可以看到
ubuntu配置环境 || 虚拟环境中【快速】安装pytorch等cudatoolkit相关环境官网已经提供了安装pytorch的命令
conda install pytorch torchvision cudatoolkit=10.2 -c pytorch
理论上只要执行者命令,就可以安装成功。可以安装的过程中发现速度很慢

2.1 安装速度奇慢
  • 原因
    使用该命令,速度非常慢,很久以后没有反应,或者失去响应了。这是因为该命令使用的是国外的源导致的。
    先解释下commod:conda install pytorch torchvision cudatoolkit=10.2 -c pytorch。
    这个命令,会安装pytorch、torchvision、cudatoolkit。后面的-c pytorch参数指定了conda获取pytorch的channel,再次指定为conda自带的pytorch仓库。

  • 解决
    -c pytorch语句去掉,然后添加清华镜像的源后,就可以使用清华镜像源快速安装pytorch了
    .
    conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
    conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
    conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
    conda config --set show_channel_urls yes
    conda install pytorch torchvision cudatoolkit=10.2

.
运行该命令,会安装很多包,等待安装结束即可。前面已经添加了清华镜像源,所以这里库的下载快很多。或者遇到这个别大包可以跳过,其他的正常安装。安装包比较大时,容易与网站失去连接,所以建议手动下载:安装过程中,在终端显示的安装包的下载链接,可以找到未成功安装的包的链接,然后访问并下载相应的包,然后进行安装即可

2.2 安装指定版本的pytorch

上面官网截图的左下角的链接,进入会发现安装指定版本的命令,关于安装速度上的问题,参考2.1的讲解即可。
.
另外要注意:要安装numpy,否则 import torch 会报错

相关文章:

  • 2022-12-23
  • 2022-12-23
  • 2021-05-23
  • 2021-08-09
  • 2021-07-10
  • 2022-12-23
  • 2022-01-02
  • 2022-12-23
猜你喜欢
  • 2022-01-19
  • 2022-12-23
  • 2021-05-07
  • 2022-12-23
  • 2021-10-21
  • 2022-12-23
  • 2022-12-23
相关资源
相似解决方案