文章目录
- 1 Linux下激活自己的虚拟环境并查看Python版本
- 2 查看需要安装的Pytorch和Torchvision版本
- 3 直接命令行安装
- 3.1 如果不报错的话
- 3.2 ERROR: Could not install packages due to an EnvironmentError: [Errno 28] No space left on device
- 4 使用安装包本地安装
- 4.1本地安装原因
- 4.2 从哪下载安装包
- 4.3 安装
- 4.4 测试是否安装成功
1 Linux下激活自己的虚拟环境并查看Python版本
如何创建虚拟环境见virtualenv创建虚拟环境
source myenv/bin/activate
激活环境后查看自己的python版本
python -V
2 查看需要安装的Pytorch和Torchvision版本
根据CUDA版本确定需要安装的Pytorch和Torchvision版本
https://pytorch.org/get-started/previous-versions/ # 官网,可能需要科学上网
3 直接命令行安装
3.1 如果不报错的话
以安装torch1.6.0为例
pip install torch==1.6.0 torchvision==0.7.0 -i https://pypi.tuna.tsinghua.edu.cn/simple/
pip install torch== -i https://pypi.tuna.tsinghua.edu.cn/simple/
3.2 ERROR: Could not install packages due to an EnvironmentError: [Errno 28] No space left on device
错误原因:
- pip安装时需要将安装包首先解压到某个位置,默认目录是 /tmp,当默认目录无空间时就会出现该问题。
- 缓存空间不足
解决方案:
TMPDIR=/data/ pip install --cache-dir=/data/ --build /data/ torch==1.10.0
4 使用安装包本地安装
4.1本地安装原因
- 命令行安装网速原因报错
- pip源上找不到期待版本安装包,换源也行不通
4.2 从哪下载安装包
打开如下网址,其中有torch和torchvision安装包
https://download.pytorch.org/whl/torch_stable.html
根据自己的CUDA版本、Python版本、操作系统选择对应的安装包进行下载
图中cu101表示cuda10.1,cp36表示python3.6,linux表示linux操作系统
4.3 安装
在激活的虚拟环境中进行安装pytorch和torchvision
pip install torch-1.6.0-cp36-cp36m-linux_x86_64.whl
pip install torchvision-0.7.0-cp36-cp36m-linux_x86_64.whl
命令行中成功安装
4.4 测试是否安装成功
在python中运行如下代码
import torch
import torchvision
print(torch.__version__)
print(torch.cuda.is_available())
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)