linux下安装torch
时间: 2023-05-01 10:04:30 浏览: 622
torch1.0.0安装
5星 · 资源好评率100%
在Linux上安装Torch可以参考以下步骤:
1. 确认你的Linux操作系统版本和架构(例如Ubuntu 18.04 64位)。
2. 安装依赖库,包括Python(建议安装Anaconda),GCC(C++编译工具),以及其他相关库,可以使用以下命令:
sudo apt-get update
sudo apt-get install python python-pip python-dev python-virtualenv python-setuptools python-numpy python-scipy python-pandas python-matplotlib python-sklearn python-skimage python-protobuf python-h5py python-leveldb python-networkx python-nose python-tk libopenblas-dev liblapack-dev libhdf5-serial-dev protobuf-compiler libleveldb-dev libsnappy-dev libopencv-dev liblmdb-dev libgflags-dev libgoogle-glog-dev libatlas-base-dev git curl wget build-essential cmake
3. 安装Torch,可以使用以下命令:
git clone https://github.com/torch/distro.git ~/torch --recursive
cd ~/torch; bash install-deps;
./install.sh
4. 安装Torch软件包,例如torchvision和cudnn,可以使用以下命令:
luarocks install torch
luarocks install cutorch
luarocks install cunn
luarocks install cudnn
5. 测试Torch是否安装成功,可以执行以下命令:
th -e "print('Hello Torch!')"
如果成功安装,将会输出"Hello Torch!"。
注意:在安装Torch之前,最好使用conda创建一个独立的虚拟环境,以免与其他Python环境冲突。
阅读全文