📜  为 cuda 10.0 安装 pytorch - Shell-Bash 代码示例

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

代码示例2
# CUDA 9.2
conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=9.2 -c pytorch

# CUDA 10.0
conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0 -c pytorch

# CPU Only
conda install pytorch==1.2.0 torchvision==0.4.0 cpuonly -c pytorch