Skip to content

Commit

Permalink
Fixup TorchCUDA
Browse files Browse the repository at this point in the history
  • Loading branch information
stemann committed Feb 22, 2022
1 parent 1802af7 commit 2fc4de7
Showing 1 changed file with 0 additions and 7 deletions.
7 changes: 0 additions & 7 deletions T/Torch/TorchCUDA/build_tarballs.jl
Original file line number Diff line number Diff line change
Expand Up @@ -28,7 +28,6 @@ mkdir -p $includedir $libdir $prefix/share
cd $WORKSPACE/srcdir
mv cudnn $prefix
if [[ $target == *linux* ]]; then
cd $bb_full_target
else
Expand All @@ -41,18 +40,12 @@ rm -r libtorch/share
mv libtorch/* $prefix
rm -r libtorch
mkdir -p /usr/local/cuda/lib64
cd /usr/local/cuda/lib64
ln -s ${prefix}/cuda/lib64/libcudart.so libcudart.so
ln -s ${prefix}/cuda/lib64/libnvToolsExt.so libnvToolsExt.so
cd $WORKSPACE/srcdir/Torch.jl/build
mkdir build && cd build
cmake -DCMAKE_PREFIX_PATH=$prefix -DTorch_DIR=$prefix/share/cmake/Torch -DCUDA_TOOLKIT_ROOT_DIR=$prefix/cuda ..
cmake --build .
cp -r $WORKSPACE/srcdir/Torch.jl/build/build/*.${dlext} "${libdir}"
rm -rf $prefix/cuda
install_license ${WORKSPACE}/srcdir/Torch.jl/LICENSE
"""

Expand Down

0 comments on commit 2fc4de7

Please sign in to comment.