Windows深度学习环境----Cuda version 10.2 pytorch3d version 0.3.0

代码写着写着就会了2024-04-08 20:59

Requirements

Python version 3.8.5
Pytorch version: pytorch==1.6.0 torchvision==0.8.2 torchaudio==0.7.0 cudatoolkit=10.2.89
pytorch3d version 0.3.0
Cuda version 10.2

感觉readme文件里的不适配，跟pytorch官网不同

复制代码

# CUDA 10.2
conda install pytorch==1.6.0 torchvision==0.7.0 cudatoolkit=10.2 -c pytorch

因为已经更换过conda源了，所以不需要添加-c pytorch -c conda-forge，否则默认从官方源添加 # 这一步需要等待一段时间，但清华源速度总比官方源强多了

提示需要 pytorch 1.7.1，于是修改

Python version 3.8.15
Pytorch version: pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=10.2.89
pytorch3d version 0.3.0
Cuda version 10.2

|-------|-------|----------------|
| 1.7.1 | 0.8.2 | >=3.6, <=3.9 |
| 1.7.0 | 0.8.1 | >=3.6, <=3.8 |
| 1.7.0 | 0.8.0 | >=3.6, <=3.8 |
| 1.6.0 | 0.7.0 | >=3.6, <=3.8 |

|-------|-------|----------------|
| 1.7.1 | 0.7.2 | >=3.6, <=3.9 |
| 1.7.0 | 0.7.0 | >=3.6, <=3.8 |
| 1.6.0 | 0.6.0 | >=3.6, <=3.8 |

PyTorch中torch、torchvision、torchaudio、torchtext版本对应关系_torch2.0.1对应的torchvision-CSDN博客

pytorch检查

下载好后可以先检查一下，在命令行依次输入：

复制代码

python
import torch
print(torch.cuda.is_available())

如果输出

复制代码

Ture

那就ok了

输入quit（）退出python