Skip to content

Commit

Permalink
small fixes
Browse files Browse the repository at this point in the history
  • Loading branch information
csukuangfj committed Oct 21, 2023
1 parent da8aa8a commit fffd5af
Show file tree
Hide file tree
Showing 2 changed files with 7 additions and 3 deletions.
8 changes: 5 additions & 3 deletions scripts/github_actions/build-ubuntu-cuda.sh
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@ fi
if [ -z $CUDA_VERSION ]; then
echo "Please set the environment variable CUDA_VERSION"
echo "Example: export CUDA_VERSION=10.2"
# valid values: 10.2, 11.1, 11.3, 11.6, 11.7, 11.8
# valid values: 10.2, 11.1, 11.3, 11.6, 11.7, 11.8, 12.1
exit 1
fi

Expand All @@ -43,6 +43,8 @@ make install >/dev/null 2>&1

popd

rm -rf Python-${PYTHON_VERSION}.3

export PATH=$PYTHON_INSTALL_DIR/bin:$PATH
export LD_LIBRARY_PATH=$PYTHON_INSTALL_DIR/lib:$LD_LIBRARY_PATH
ls -lh $PYTHON_INSTALL_DIR/lib/
Expand All @@ -63,8 +65,8 @@ python3 -m pip install bs4 requests tqdm auditwheel
echo "Installing torch"
./install_torch.sh

rm -rf ~/.cache/pip
yum clean all
rm -rf ~/.cache/pip >/dev/null 2>&1
yum clean all >/dev/null 2>&1

cd /var/www

Expand Down
2 changes: 2 additions & 0 deletions scripts/github_actions/install_torch.sh
Original file line number Diff line number Diff line change
Expand Up @@ -13,9 +13,11 @@
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
set -x

torch=$TORCH_VERSION
cuda=$CUDA_VERSION
echo "torch: $torch, cuda: $cuda"
case ${torch} in
1.5.*)
case ${cuda} in
Expand Down

0 comments on commit fffd5af

Please sign in to comment.