mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
## Summary - Replaced miniconda base image with base Ubuntu 22.04 image - Installed Python and required dependencies using apt - Replaced conda-based CUDA installation with apt-based version - Updated paths in install-dev-tools.sh to reflect the new non-conda environment - Removed conda-specific files and added requirements.txt for Python dependencies ## Test plan Test by building and running the devcontainer in VS Code with both CPU and CUDA configurations Pull Request resolved: https://github.com/pytorch/pytorch/pull/152881 Approved by: https://github.com/atalman
14 lines
504 B
Bash
14 lines
504 B
Bash
#!/usr/bin/env bash
|
|
# Run this command from the PyTorch directory after cloning the source code using the “Get the PyTorch Source“ section below
|
|
pip install -r requirements.txt
|
|
git submodule sync
|
|
git submodule update --init --recursive
|
|
|
|
# This takes some time
|
|
make setup-lint
|
|
|
|
# Add CMAKE_PREFIX_PATH to bashrc
|
|
echo 'export CMAKE_PREFIX_PATH=/usr/local' >> ~/.bashrc
|
|
# Add linker path so that cuda-related libraries can be found
|
|
echo 'export LDFLAGS="-L/usr/local/cuda/lib64/ $LDFLAGS"' >> ~/.bashrc
|