pytorch_lighting的一些环境设置问题

pytorch版本要和自己所加载的包版本对应，不然会很麻烦

我在跑代码的时候是遇到了：Cannot import pytorch_lightning module: 'torch' has no attribute 'autocast'这个问题，想了一下原因是版本没有对应的上。然后再知乎上找到了这个：

PyTorch、PyG 和 PyTorch Lightning 版本问题（QCNet环境配置问题）我具体是参考的下面这个知乎

PyTorch、PyG 和 PyTorch Lightning 版本问题（QCNet环境配置问题） - 知乎

因为我总是重装pytorch所以在这里附上直接安装的代码。这个是我的具体版本 cuda10.1和pytorch1.7.1

pip install torch==1.7.1+cu101 torchvision==0.8.2+cu101 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html

验证是否成功装好的代码

import torch

# 检查是否安装了 CUDA 版本的 PyTorch
print("PyTorch CUDA availability:", torch.cuda.is_available())

# 如果 CUDA 可用，输出 CUDA 版本信息
if torch.cuda.is_available():
    print("CUDA version:", torch.version.cuda)


# 检查是否有可用的 GPU
if torch.cuda.is_available():
    print("GPU 可用")
else:  
    print("GPU 不可用")

# 打印 PyTorch 版本
print("PyTorch 版本:", torch.__version__)