Sure, here are the steps to install Torch with CUDA 11:
- Make sure you have CUDA 11 installed on your system. You can download the CUDA 11 installer from the NVIDIA website.
- Install the dependencies needed for Torch. Run the following commands in your terminal:
sudo apt-get update
sudo apt-get install build-essential curl git cmake
sudo apt-get install libcurl4-openssl-dev libssl-dev zlib1g-dev
Download the Torch installer from the official website using the following command:
curl -s https://raw.githubusercontent.com/torch/distro/master/install-deps | bash
Install Torch with CUDA 11 support by running the following command:
git clone https://github.com/torch/distro.git ~/torch --recursive
cd ~/torch; bash install.sh -c cuda-11.0
Wait for the installation process to complete. Once done, you can test the installation by running the following command:
th -e "require 'cutorch'; print(cutorch.getDeviceCount())"
This should output the number of available GPUs on your system. If it does, then Torch with CUDA 11 has been successfully installed.
