打飞垃圾袋 发表于 2023-4-8 11:24:19

怎样3060运行cuda10环境下的代码

我下载了PCRNet代码,他的运行环境如图所示,我在网上查到cuda不支持cuda10。我应该怎样修改代码或者配置环境才能正常运行代码
Requirements:
[*]Cuda 10
[*]pytorch==1.4.0
[*]transforms3d==0.3.1
[*]h5py==2.9.0

isdkz 发表于 2023-4-8 11:28:55

如果您的NVIDIA显卡是RTX 3060,那么您应该使用支持该显卡的较新版本的CUDA。为了在较新版本的CUDA环境下运行该代码,您可以尝试以下步骤:

1、首先升级CUDA。建议安装CUDA 11.x,因为它支持NVIDIA RTX 3060显卡。您可以在NVIDIA官网上找到安装CUDA的教程:https://developer.nvidia.com/cuda-toolkit

2、安装与新版本CUDA兼容的PyTorch。您可以访问PyTorch官方网站来查找支持CUDA 11.x的PyTorch版本,按照提示安装:https://pytorch.org/get-started/locally/

例如,假设您已经安装了CUDA 11.1,那么您可以使用以下命令安装PyTorch(请根据您的实际情况修改CUDA版本和PyTorch版本):

pip install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/cu111/torch_stable.html

3、安装其他依赖库。您可以继续使用PCRNet代码所需的其他依赖库的版本。安装这些库的命令如下:

pip install transforms3d==0.3.1 h5py==2.9.0

4、修改代码以适应新环境。尽管升级了CUDA和PyTorch版本,但在大多数情况下,代码应该能够正常运行。如果您在运行代码时遇到兼容性问题,您可能需要根据错误提示修改代码。针对PyTorch的兼容性问题,您可以查阅PyTorch的迁移指南:https://pytorch.org/docs/stable/index.html

5、最后,运行代码。在完成上述步骤后,您应该可以在新环境下正常运行PCRNet代码了。如果遇到其他问题,请根据错误提示进行调试。

希望这些建议能对您有所帮助!
页: [1]
查看完整版本: 怎样3060运行cuda10环境下的代码