mirror of
https://github.com/pjreddie/darknet.git
synced 2023-08-10 21:13:14 +03:00
move setup scripts to scripts subfolder
This commit is contained in:
40
scripts/setup.ps1
Executable file
40
scripts/setup.ps1
Executable file
@ -0,0 +1,40 @@
|
||||
## enable or disable installed components
|
||||
|
||||
$install_cuda=$false
|
||||
$vcpkg_folder=".\"
|
||||
$temp_folder=".\temp"
|
||||
|
||||
###########################
|
||||
|
||||
New-Item -Path . -Name $temp_folder -ItemType "directory"
|
||||
Set-Location $temp_folder
|
||||
|
||||
# Download and install Chocolatey
|
||||
Set-ExecutionPolicy unrestricted
|
||||
Invoke-Expression ((New-Object System.Net.WebClient).DownloadString('https://chocolatey.org/install.ps1'))
|
||||
choco.exe install -y cmake ninja powershell git vscode
|
||||
choco-exe install -y visualstudio2019buildtools --package-parameters "--add Microsoft.VisualStudio.Component.VC.CoreBuildTools --includeRecommended --includeOptional --passive --locale en-US --lang en-US"
|
||||
|
||||
if ($install_cuda) {
|
||||
# Download and install CUDA
|
||||
Invoke-WebRequest https://developer.nvidia.com/compute/cuda/10.1/Prod/local_installers/cuda_10.1.105_418.96_win10.exe -OutFile .\cuda_setup.exe
|
||||
.\cuda_setup.exe -s nvcc_10.1 cuobjdump_10.1 nvprune_10.1 cupti_10.1 gpu_library_advisor_10.1 memcheck_10.1 nvdisasm_10.1 nvprof_10.1 visual_profiler_10.1 visual_studio_integration_10.1 cublas_10.1 cublas_dev_10.1 cudart_10.1 cufft_10.1 cufft_dev_10.1 curand_10.1 curand_dev_10.1 cusolver_10.1 cusolver_dev_10.1 cusparse_10.1 cusparse_dev_10.1 nvgraph_10.1 nvgraph_dev_10.1 npp_10.1 npp_dev_10.1 nvrtc_10.1 nvrtc_dev_10.1 nvml_dev_10.1 occupancy_calculator_10.1 fortran_examples_10.1
|
||||
|
||||
$env:CUDA_PATH = "${env:ProgramFiles}\NVIDIA GPU Computing Toolkit\CUDA\v10.1"
|
||||
$env:CUDA_PATH_V10_1 = $env:CUDA_PATH
|
||||
$env:CUDA_TOOLKIT_ROOT_DIR = $env:CUDA_PATH
|
||||
$env:PATH += ";${env:CUDA_PATH}\bin;"
|
||||
|
||||
$features = "full"
|
||||
}
|
||||
else {
|
||||
$features = "opencv-base,weights,weights-train"
|
||||
}
|
||||
|
||||
Remove-Item -r $temp_folder
|
||||
Set-Location ..
|
||||
Set-Location $vcpkg_folder\
|
||||
git.exe clone https://github.com/Microsoft/vcpkg
|
||||
Set-Location vcpkg
|
||||
.\bootstrap-vcpkg.bat
|
||||
.\vcpkg.exe install darknet[${features}]:x64-windows
|
48
scripts/setup.sh
Executable file
48
scripts/setup.sh
Executable file
@ -0,0 +1,48 @@
|
||||
#!/usr/bin/env bash
|
||||
|
||||
## enable or disable installed components
|
||||
|
||||
install_cuda=false
|
||||
vcpkg_folder="."
|
||||
temp_folder="./temp"
|
||||
|
||||
###########################
|
||||
|
||||
mkdir $temp_folder
|
||||
cd $temp_folder
|
||||
|
||||
sudo apt-get install cmake git ninja-build build-essentials g++
|
||||
|
||||
if [ "$install_cuda" = true ] ; then
|
||||
if [[ "$OSTYPE" == "darwin"* ]]; then
|
||||
echo "Unable to provide CUDA on macOS"
|
||||
else
|
||||
# Download and install CUDA
|
||||
wget http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1404/x86_64/cuda-repo-ubuntu1404_10.0.130-1_amd64.deb
|
||||
sudo apt-key adv --fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1404/x86_64/7fa2af80.pub
|
||||
sudo dpkg -i cuda-repo-ubuntu1404_10.0.130-1_amd64.deb
|
||||
wget http://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1404/x86_64/nvidia-machine-learning-repo-ubuntu1404_4.0-2_amd64.deb
|
||||
sudo dpkg -i nvidia-machine-learning-repo-ubuntu1404_4.0-2_amd64.deb
|
||||
sudo apt-get -y update
|
||||
sudo apt-get install -y --no-install-recommends cuda-compiler-10-0 cuda-libraries-dev-10-0 cuda-driver-dev-10-0 cuda-cudart-dev-10-0 cuda-cublas-dev-10-0 cuda-curand-dev-10-0
|
||||
sudo apt-get install -y --no-install-recommends libcudnn7-dev
|
||||
sudo ln -s /usr/local/cuda-10.0/lib64/stubs/libcuda.so /usr/local/cuda-10.0/lib64/stubs/libcuda.so.1
|
||||
|
||||
export CUDACXX=/usr/local/cuda-10.0/bin/nvcc
|
||||
export CUDA_PATH=/usr/local/cuda-10.0
|
||||
export CUDA_TOOLKIT_ROOT_DIR=/usr/local/cuda-10.0
|
||||
export LD_LIBRARY_PATH="/usr/local/cuda-10.0/lib64:/usr/local/cuda-10.0/lib64/stubs:${LD_LIBRARY_PATH}"
|
||||
|
||||
features = "full"
|
||||
fi
|
||||
else
|
||||
features = "opencv-base,weights,weights-train"
|
||||
fi
|
||||
|
||||
rm -rf $temp_folder
|
||||
cd ..
|
||||
cd $vcpkg_folder
|
||||
git clone https://github.com/Microsoft/vcpkg
|
||||
cd vcpkg
|
||||
./bootstrap-vcpkg.sh
|
||||
./vcpkg install darknet[${features}]
|
Reference in New Issue
Block a user