This script can be used to automatically install torch and CUDA.
Project description
Install torch
This script can be used to automatically install torch and CUDA.
How it works?
- The user is prompted to select whether they want to install the CPU or GPU version of torch.
- If the GPU version is selected, the user is prompted to enter the version of CUDA they want to install.
- The script checks if CUDA is already installed and if it is not, it downloads and installs the version specified by the user.
- The script then downloads and installs the appropriate version of torch based on the user's initial selection.
Requirements
This script requires the following Python packages:
- rich
- beautifulsoup4
- wget
- requests
- requests-html
- tqdm
You can install these packages by running pip install -r requirements.txt
.
Usage
To use this script install it using pip:
pip install install_torch
install-torch
Alternatively, you can download the script and run it directly using Python:
git clone https://github.com/hawier-dev/install_torch.git
python install_torch.py
License
This project is licensed under the MIT License - see the LICENSE file for details.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
install_torch-1.0.tar.gz
(3.5 kB
view hashes)
Built Distribution
Close
Hashes for install_torch-1.0-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 9ca5b4d80a34e6372c90afd6fd344913f818fdad0cd78577f12505b3a01ef8d2 |
|
MD5 | ea428e8385df67fdb2a90d367e2a37d3 |
|
BLAKE2b-256 | ec6bb30715b9f5a9d45103ec999d0a5c3e5cb50f635a376b93e64a049b09f86e |