mirror of
https://github.com/facebookresearch/pytorch3d.git
synced 2025-08-01 11:22:49 +08:00
wheels with cuda
Summary: The main pytorch wheels on PyPI support CUDA 10.2. Here we make pytorch3d's wheels do the same, instead of being cpu only. This should ultimately make life easier in colab. Also a little script to count builds, which can be useful for nightly jobs. Reviewed By: gkioxari Differential Revision: D22924321 fbshipit-source-id: d6cea9bfbab49bcb0080f65608066c553ea8bb4d
This commit is contained in:
parent
07d7e12644
commit
4872a2c4de
29
.circleci/build_count.py
Normal file
29
.circleci/build_count.py
Normal file
@ -0,0 +1,29 @@
|
||||
#!/usr/bin/env python3
|
||||
# Copyright (c) Facebook, Inc. and its affiliates. All rights reserved.
|
||||
|
||||
"""
|
||||
Print the number of nightly builds
|
||||
"""
|
||||
|
||||
from collections import Counter
|
||||
|
||||
import yaml
|
||||
|
||||
|
||||
conf = yaml.safe_load(open("config.yml"))
|
||||
jobs = conf["workflows"]["build_and_test"]["jobs"]
|
||||
|
||||
|
||||
def jobtype(job):
|
||||
if isinstance(job, str):
|
||||
return job
|
||||
if len(job) == 1:
|
||||
[name] = job.keys()
|
||||
return name
|
||||
return "MULTIPLE PARTS"
|
||||
|
||||
|
||||
for i, j in Counter(map(jobtype, jobs)).items():
|
||||
print(i, j)
|
||||
print()
|
||||
print(len(jobs))
|
@ -55,7 +55,7 @@ binary_common: &binary_common
|
||||
wheel_docker_image:
|
||||
description: "Wheel only: what docker image to use"
|
||||
type: string
|
||||
default: "pytorch/manylinux-cuda101"
|
||||
default: "pytorch/manylinux-cuda102"
|
||||
environment:
|
||||
PYTHON_VERSION: << parameters.python_version >>
|
||||
BUILD_VERSION: << parameters.build_version >>
|
||||
|
@ -55,7 +55,7 @@ binary_common: &binary_common
|
||||
wheel_docker_image:
|
||||
description: "Wheel only: what docker image to use"
|
||||
type: string
|
||||
default: "pytorch/manylinux-cuda101"
|
||||
default: "pytorch/manylinux-cuda102"
|
||||
environment:
|
||||
PYTHON_VERSION: << parameters.python_version >>
|
||||
BUILD_VERSION: << parameters.build_version >>
|
||||
@ -386,18 +386,18 @@ workflows:
|
||||
python_version: '3.8'
|
||||
pytorch_version: 1.6.0
|
||||
- binary_linux_wheel:
|
||||
cu_version: cpu
|
||||
name: linux_wheel_py36_cpu_pyt160
|
||||
cu_version: cu102
|
||||
name: linux_wheel_py36_cu102_pyt160
|
||||
python_version: '3.6'
|
||||
pytorch_version: 1.6.0
|
||||
- binary_linux_wheel:
|
||||
cu_version: cpu
|
||||
name: linux_wheel_py37_cpu_pyt160
|
||||
cu_version: cu102
|
||||
name: linux_wheel_py37_cu102_pyt160
|
||||
python_version: '3.7'
|
||||
pytorch_version: 1.6.0
|
||||
- binary_linux_wheel:
|
||||
cu_version: cpu
|
||||
name: linux_wheel_py38_cpu_pyt160
|
||||
cu_version: cu102
|
||||
name: linux_wheel_py38_cu102_pyt160
|
||||
python_version: '3.8'
|
||||
pytorch_version: 1.6.0
|
||||
- binary_linux_conda_cuda:
|
||||
|
@ -38,7 +38,7 @@ def workflows(prefix="", filter_branch=None, upload=False, indentation=6):
|
||||
)
|
||||
for btype in ["wheel"]:
|
||||
for python_version in ["3.6", "3.7", "3.8"]:
|
||||
for cu_version in ["cpu"]:
|
||||
for cu_version in ["cu102"]:
|
||||
w += workflow_pair(
|
||||
btype=btype,
|
||||
python_version=python_version,
|
||||
|
Loading…
x
Reference in New Issue
Block a user