PyTorch——配置深度学习环境

151 阅读3分钟

还在为各种包的版本匹配发愁,还在冥想这个错误到底从何而来。
放弃吧,重装(换个环境)才是王道。

安装Cuda

tensor在GPU上运行是可以提高一定的处理效果的 首先需要检查一下你的GPU,查看一下版本 很简单进入任务管理器,打开性能,就能看见你GPU的参数啦(一般游戏笔记本是有两张显卡的,一张独显,一张集显)

img

前往Cuda官网 下载驱动,安装即可

img

下载后,默认安装就好。然后打开cmd,输入 nvidia-smi 即可查看

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ 
conda config --add channels http://mirrors.ustc.edu.cn/anaconda/pkgs/free/
conda config --set show_channel_urls yes

img

安装Anaconda

硬件好了,我们需要一个集成环境帮我们管理包和他们之间的依赖 anaconda当然是很好的选择

这里推荐两个路径:Anaconda官网清华软件镜像园

同样也是下载安装即可,注意勾选默认到Path环境变量就行。

这里我们需要注意,修改一下anaconda的源:清华大学开源软件镜像站和中国科技大学镜像站 方便我们下载

conda config --add channels http://mirrors.ustc.edu.cn/anaconda/pkgs/free/
conda config --add channels http://mirrors.ustc.edu.cn/anaconda/pkgs/main/
conda config --add channels http://mirrors.ustc.edu.cn/anaconda/cloud/menpo/
conda config --add channels http://mirrors.ustc.edu.cn/anaconda/cloud/bioconda/
conda config --add channels http://mirrors.ustc.edu.cn/anaconda/cloud/msys2/
conda config --add channels http://mirrors.ustc.edu.cn/anaconda/cloud/conda-forge/
​
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/ 
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/ 
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/peterjc123/
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/    #安装pytorch的小伙伴这个源一定不要漏
​
conda config --set show_channel_urls yes

衷心提醒网址一定要是http 不能是https,要不然会访问报错。 网址也不需要加引号切记

当然也可以自己手动去 C:用户下修改.condarc(这里贴出我的文件内容)

ssl_verify: true
channels:
  - defaults
show_channel_urls: true
channel_alias: https://mirrors.tuna.tsinghua.edu.cn/anaconda
default_channels:
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/pro
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/peterjc123/
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
custom_channels:
  conda-forge: http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  msys2: http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  bioconda: http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  menpo: http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  pytorch: http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  simpleitk: http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud

查看conda版本 conda -version

升级conda工具包 conda upgrade --all

查看当前conda环境 conda env list

创建虚拟环境 conda create -n python=3.8 这样anaconda就会去找python对应的包 耐心等待就好

激活环境 conda activate py38 我这里的py38就是我新建的虚拟环境名称

conda还有一个好处 就是可以版本回滚 方便回到之间的状态

conda list --revisions

img

若想回滚到某个时刻,直接输入那个版本号就行了,例如我想回滚到第10个版本,即

conda install --revision 10

安装torch

直接在官网寻找就好了,对应好自己的操作平台,和cuda版本下载即可

img

之前设置的源,会为我们找到国内的镜像路径,加快下载速度

测试torch

如果在Pycharm内,直接进入project选择新建虚拟环境的python.exe即可 如果在Jupyter Notebook中,则需要多安装一个包 nb_conda 安装完以后你就会发现你的jupyter会多一个conda插卡,可以查看conda的虚拟环境

img

运行的程序的时候,注意选择kernel是哪个环境的 以免发生版本错误 这些都是显示torch信息的一些命令可以用于测试的哦

import torch
print(torch.__version__)
torch.cuda.is_available()
torch.cuda.device_count()
torch.cuda.get_device_name(0)