📜  火炬 1.1.0 安装 - Shell-Bash 代码示例

📅  最后修改于: 2022-03-11 14:50:04.645000             🧑  作者: Mango

代码示例1
# CUDA 9.0
conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=9.0 -c pytorch

# CUDA 10.0
conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=10.0 -c pytorch

# CPU Only
conda install pytorch-cpu==1.1.0 torchvision-cpu==0.3.0 cpuonly -c pytorch