问题解决(2)| 关于pytorch安装的一些事儿

1.使用官网命令安装pytorch

官网地址:https://pytorch.org/get-started/previous-versions/(因为我一直是安装旧版本)

这里只提供1.6.0的安装方法:

v1.6.0

Conda

OSX

# conda
conda install pytorch==1.6.0 torchvision==0.7.0 -c pytorch

Linux and Windows

# CUDA 9.2
conda install pytorch==1.6.0 torchvision==0.7.0 cudatoolkit=9.2 -c pytorch

# CUDA 10.1
conda install pytorch==1.6.0 torchvision==0.7.0 cudatoolkit=10.1 -c pytorch

# CUDA 10.2
conda install pytorch==1.6.0 torchvision==0.7.0 cudatoolkit=10.2 -c pytorch

# CPU Only
conda install pytorch==1.6.0 torchvision==0.7.0 cpuonly -c pytorch

Wheel

OSX

pip install torch==1.6.0 torchvision==0.7.0

Linux and Windows

# CUDA 10.2
pip install torch==1.6.0 torchvision==0.7.0

# CUDA 10.1
pip install torch==1.6.0+cu101 torchvision==0.7.0+cu101 -f https://download.pytorch.org/whl/torch_stable.html

# CUDA 9.2
pip install torch==1.6.0+cu92 torchvision==0.7.0+cu92 -f https://download.pytorch.org/whl/torch_stable.html

# CPU only
pip install torch==1.6.0+cpu torchvision==0.7.0+cpu -f https://download.pytorch.org/whl/torch_stable.html

1.1 如何查看自己机器的cuda版本

Windows

因为我这台机器没有显卡,所以转载别人的:

方法一

打开cmd,输入:

nvcc --version
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2019 NVIDIA Corporation
Built on Sun_Jul_28_19:12:52_Pacific_Daylight_Time_2019
Cuda compilation tools, release 10.1, V10.1.243 ##这里就是版本号

方法二
(1)、按win+Q,输入NVIDIA Control Panel 打开NVIDIA控制面板

在这里插入图片描述

再点系统信息
在这里插入图片描述
点击组件,就可以看到你的CUDA信息

在这里插入图片描述

Linux

打开Linux服务器,输入以下命令:

cat /usr/local/cuda/version.txt

可以观察到:

CUDA Version 10.2.89

2.pip安装pytorch出现MemoryError

pip安装pytorch出现MemoryError如下图:

根据最后一行红色字体我们知道出现了MemoryError,根据字面意思我们可知此问题与内存有关。
因为pip安装的缓存机制想要先把整个文件读取到内存以后才开始安装,因此可能导致内存不足。所以我们在安装时指示不启用缓存即可,可以使用 --no-cache-dir 命令,如下

pip install --no-cache-dir torch==1.5.1+cu101 torchvision==0.6.1+cu101 -f https://download.pytorch.org/whl/torch_stable.html

 3.from torchcrf import CRF 如何安装torchcrf

 

安装torchcrf

错误1: pip install torchcrf
错误2: pip install pytorch-crf==0.4.0
解决:第二个安装后需要先卸载:(没安装过可跳过这一步)
pip uninstall pytorch-crf==0.4.0
然后: pip install pytorch-crf

 

4.Anaconda安装及添加清华Pytorch镜像

  • Anaconda安装

    bash Anaconda2-4.4.0-Linux-x86_64.sh

     

  • 添加清华镜像

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 --set show_channel_urls yes
  • 添加Pytorch镜像

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

# for legacy win-64
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/peterjc123/

 

 

 

 

 

 

 

 

  • 1
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值