解决cuda、torchtext、torchvision和pytorch的版本对应问题

2022.5.5 更新

今天官网里看到已经不支持10.2版本的CUDA下载,
实验室电脑GPU为GeForce GT 730,算力太低,索性换成cpu版本的pytorch


给出torch的版本下载和torch工具的下载地址:
http://download.pytorch.org/whl/torch_stable.html
https://pytorch-geometric.com/whl/torch-1.10.0%2Bcu102.html
记得要对应!对应!还是tmd对应!

cuda和pytorch的版本对应如下:
解决cuda、torchtext、torchvision和pytorch的版本对应问题_第1张图片

附上torchvision和pytorch的版本对应关系:
https://github.com/pytorch/vision#installation
解决cuda、torchtext、torchvision和pytorch的版本对应问题_第2张图片

一般有两种安装方式,一个是从anaconda安装,另一个直接在命令行窗口安装。有时候anaconda不太稳定,也建议直接下载下来后在终端pip安装。
附上pytorch的下载地址:http://download.pytorch.org/whl/torch_stable.html
Anaconda:

# 直接安装对应pytorch版本的torchvision
conda install torchvision -c pytorch

pip:

pip install torchvision

你可能感兴趣的:(python,pytorch)