Skip to content
Snippets Groups Projects
Commit 092ef7a3 authored by Ludovic Noels's avatar Ludovic Noels
Browse files
parents 6966bfe0 9ff7d75c
No related branches found
No related tags found
1 merge request!309Master
...@@ -310,8 +310,16 @@ void Tree::createRandomTreeSameDepthForNodes(int nbLeaves, int numChildMax, int ...@@ -310,8 +310,16 @@ void Tree::createRandomTreeSameDepthForNodes(int nbLeaves, int numChildMax, int
} }
} }
else else
{
if (numChilds+remaining < 4)
{ {
outPut.push_back(numChilds+remaining); outPut.push_back(numChilds+remaining);
}
else
{
outPut.push_back(2);
outPut.push_back(numChilds+remaining-2);
}
iter += numChilds+remaining; iter += numChilds+remaining;
} }
}; };
......
...@@ -519,16 +519,14 @@ D) Install Torch ...@@ -519,16 +519,14 @@ D) Install Torch
4) install cudnn 4) install cudnn
download cudnn from nvidia (you need to register and downloadmanualy download cudnn from nvidia (you need to register and downloadmanualy
https://developer.nvidia.com/compute/machine-learning/cudnn/secure/8.0.4/11.0_20200923/cudnn-11.0-linux-x64-v8.0.4.30.tgz https://developer.nvidia.com/compute/machine-learning/cudnn/secure/8.0.4/11.0_20200923/cudnn-11.0-linux-x64-v8.0.4.30.tgz
copy in your local directory
tar xvf cudnn-11.0-linux-x64-v8.0.4.30.tgz
sudo cp cuda/include/cudnn*.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
A sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*
A 5) download torch with gpu
sudo apt-get install nvidia-cuda-toolkit
//check version nvcc --version
wget https://download.pytorch.org/libtorch/cu110/libtorch-shared-with-deps-1.7.0%2Bcu110.zip wget https://download.pytorch.org/libtorch/cu110/libtorch-shared-with-deps-1.7.0%2Bcu110.zip
unzip libtorch-shared-with-deps-1.7.0%2Bcu110.zip
add export TORCHDIR=$HOME/local/libtorch in your .bashrc add export TORCHDIR=$HOME/local/libtorch in your .bashrc
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment