快速安装Pytorch和Torchvision
文章目录1 Linux下激活自己的虚拟环境并查看Python版本2 查看需要安装的Pytorch和Torchvision版本3 直接命令行安装4 使用安装包本地安装4.1本地安装原因4.2 从哪下载安装包4.3 安装4.4 测试是否安装成功1 Linux下激活自己的虚拟环境并查看Python版本如何创建虚拟环境见virtualenv创建虚拟环境source myenv/bin/activate激活
·
文章目录
1 Linux下激活自己的虚拟环境并查看Python版本
如何创建虚拟环境见virtualenv创建虚拟环境
source myenv/bin/activate
激活环境后查看自己的python版本
python -V # 我的是python3.6
2 查看需要安装的Pytorch和Torchvision版本
根据CUDA版本确定需要安装的Pytorch和Torchvision版本
https://pytorch.org/get-started/previous-versions/ # 官网,可能需要科学上网
3 直接命令行安装
3.1 如果不报错的话
以安装torch1.6.0为例
# -i 后面表示pip镜像源,不用镜像源太慢,本地以清华源为例,可更换
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())
更多推荐
已为社区贡献1条内容
所有评论(0)