conda install pytorch==1.0.0 torchvision==0.2.1 -c pytorch 和conda install pytorch==1.0.0 torchvision==0.2.1 cuda90 -c pytorch有什么区别
时间: 2023-09-16 08:07:05 浏览: 291
浅谈pytorch、cuda、python的版本对齐问题
5星 · 资源好评率100%
`conda install pytorch==1.0.0 torchvision==0.2.1 -c pytorch` 是安装 PyTorch 和 torchvision 的 CPU 版本,而 `conda install pytorch==1.0.0 torchvision==0.2.1 cuda90 -c pytorch` 是安装 PyTorch 和 torchvision 的 CUDA 版本,需要与 NVIDIA 的 GPU 驱动程序和 CUDA 工具包一起使用。如果你不确定你的 GPU 是否支持 CUDA,可以使用第一种方式安装 CPU 版本。
阅读全文