torch环境各个版本相互支持情况
pip报错,首先检查源的问题去用户目录下conda文件删除一些内容
官方推荐方式,但是这种方式由于墙的存在容易挂掉
cuda
版本依赖关系
注意:>=516.01指的是你的显卡驱动版本517.0、518.0都可以安装CUDA11.7GA版本的CUDA
torch
whl下载链接,下载完成直接pip install ***.whl
注意:aarch64表示arm64位适用于树莓派和Xavier的ARM架构的CPU
测试输出为true表示配置成功
import torch
print(torch.cuda.is_available())
本博客所有文章除特别声明外,均采用 CC BY-NC-SA 4.0 许可协议。转载请注明来自 爱吃糖的猫!
评论