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/