small fixes

This commit is contained in:
Fangjun Kuang 2023-10-21 23:49:42 +08:00
parent da8aa8a137
commit fffd5af574
2 changed files with 7 additions and 3 deletions

View File

@ -18,7 +18,7 @@ fi
if [ -z $CUDA_VERSION ]; then if [ -z $CUDA_VERSION ]; then
echo "Please set the environment variable CUDA_VERSION" echo "Please set the environment variable CUDA_VERSION"
echo "Example: export CUDA_VERSION=10.2" 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 exit 1
fi fi
@ -43,6 +43,8 @@ make install >/dev/null 2>&1
popd popd
rm -rf Python-${PYTHON_VERSION}.3
export PATH=$PYTHON_INSTALL_DIR/bin:$PATH export PATH=$PYTHON_INSTALL_DIR/bin:$PATH
export LD_LIBRARY_PATH=$PYTHON_INSTALL_DIR/lib:$LD_LIBRARY_PATH export LD_LIBRARY_PATH=$PYTHON_INSTALL_DIR/lib:$LD_LIBRARY_PATH
ls -lh $PYTHON_INSTALL_DIR/lib/ ls -lh $PYTHON_INSTALL_DIR/lib/
@ -63,8 +65,8 @@ python3 -m pip install bs4 requests tqdm auditwheel
echo "Installing torch" echo "Installing torch"
./install_torch.sh ./install_torch.sh
rm -rf ~/.cache/pip rm -rf ~/.cache/pip >/dev/null 2>&1
yum clean all yum clean all >/dev/null 2>&1
cd /var/www cd /var/www

View File

@ -13,9 +13,11 @@
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and # See the License for the specific language governing permissions and
# limitations under the License. # limitations under the License.
set -x
torch=$TORCH_VERSION torch=$TORCH_VERSION
cuda=$CUDA_VERSION cuda=$CUDA_VERSION
echo "torch: $torch, cuda: $cuda"
case ${torch} in case ${torch} in
1.5.*) 1.5.*)
case ${cuda} in case ${cuda} in