- 首先必须要安装好cuda和cudann,我安装的是如下版本的cuda
- 需要注意cuda版本和之后的pytorch版本需要匹配
- cuda10.0环境安装Pytorch_先说一下,这里不是博客,只是我的笔记本-CSDN博客_cuda10.0对应的pytorch
-
# 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 -
由于我选择的是cuda10.0,所以安装第二个版本的Pytorch
-
Pytorch官网有以下版本,但是没有cuda10.0,因此只能用上面的来安
-
pytorch的安装
于 2022-02-22 11:27:29 首次发布
本文介绍了如何在CUDA10.0环境下安装匹配的PyTorch版本。强调了CUDA版本与PyTorch版本之间的兼容性,并提供了conda安装命令。注意,PyTorch官网未直接提供CUDA10.0的版本,需要使用指定命令进行安装。
摘要由CSDN通过智能技术生成