我成功的环境:
- CUDA 11.6,cuDNN v8.8.1
python 3.9,pytorch 1.12,pytorch3d 0.7.1
Visual studio 2019
他们的版本是严格对应的,我之前CUDA 11.7,pytorch 2.0,pytorch3d 0.73安装失败了,就是因为版本没对应好
一.基础环境搭建:Anaconda+Cudnn+Cuda+Pytorch+Pycharm
参考:https://blog.csdn.net/rothschild666/article/details/121593090
提示:CUDA与CUDNN官网进入慢,可以选用中国镜像源:(把官网的com改成cn)
CUDA :https://developer.nvidia.cn/cuda-toolkit-archive
CUDNN:https://developer.nvidia.cn/rdp/cudnn-archive

本文记录了在Win10系统中安装PyTorch3D的过程,包括Anaconda环境搭建、CUDA和cuDNN的安装、VS2019配置以及PyTorch3D的源码安装。成功配置的环境为CUDA 11.6、cuDNN v8.8.1、Python 3.9、PyTorch 1.12和PyTorch3D 0.7.1。文章提供了各步骤的参考链接及注意事项,如CUDA自定义安装、VS2019环境变量设置和解决torch-scatter安装问题。

最低0.47元/天 解锁文章
7526

被折叠的 条评论
为什么被折叠?



