背景
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
没有报错就没问题了。