torch-radon2.0安装记录

背景

torch-radon库可用于解决计算机断层扫描(CT)重建问题，它被广泛应用于许多相关算法中。该库旨在帮助研究CT问题的研究人员将深度学习和基于模型的方法结合起来。需要注意的是，该库只能在Linux系统上安装哦~

---

适用的环境

该库要求Python版本为3.6或3.7或3.8，pytorch版本为1.5到1.8，cuda版本为10.1到11.1。
自测版本是否合适，如果输出的是3个ok就是合适的：

import sys
import torch
python_version = sys.version_info.major * 10 + sys.version_info.minor
if python_version in [36, 37, 38]:
    print(ok)
else:
    print(error)
if torch_available:
    torch_version = torch.__version__[:3]
    if torch_version in ["1.8", "1.7", "1.6", "1.5"]:
        print(ok)
    else:
        print(error)
else:
    print("You need to have PyTorch installed")
# CUDA
cuda_version = torch.version.cuda
if cuda_version in ["10.1", "10.2", "11.0", "11.1"]:
    print(ok)
else:
    print(error)

此外还需注意这三者版本是否匹配，以及cuda版本和你的显卡是否匹配。
cuda和pytorch的匹配
GPU 算力与 CUDA 版本对应关系

---

我的环境和安装步骤

显卡为3090Ti，pytorch 1.8.0，cuda 11.1，Python 3.8，platform: Linux-64，Ubuntu系统
安装步骤：
1.打开torch-radon的下载地址，选择download zip
2.解压压缩包，进入torch-radon文件夹并运行 auto_install.py，然后就是一长串的安装过程，一般来说只要前面的环境对了就没问题。
测试是否安装成功：

import torch
from torch_radon import Volume2D
from torch_radon.radon import FanBeam

没有报错就没问题了。