Advertisement

Linux CUDA 9 PyTorch 04 torchvision 02 machine learning (基于GPU的深度学习框架)

阅读量:

近期,作者正在深入研究图卷积网络(GCN),在查阅相关文献时发现了一篇题为《Spatial Temporal Graph Convolutional Networks for Skeleton-Based Action Recognition》的论文,内容颇具启发性。由于该论文在GitHub平台上提供了开源代码,因此决定尝试复现其研究成果。


当前所使用的服务器配置为(cuda9+cudnn7+tensorflow-gpu=1.7),该设备为多人共享资源,CUDA版本相对较为陈旧。

  • 首先,通过conda工具创建了一个专属的虚拟环境作为后续工作的基础
复制代码
    conda env list  # 可以先查看一下现有的虚拟环境
    conda create -n your_env_name python=3.6  # 创建python环境,我这里默认选择了3.6
    source activate your_env_name  # 激活创建环境,成功的话命令行最前面出现(your_env_name)
    
    
      
      
      
    

在此强烈推荐用户避免选用Python 3.7版本。起初本人随意选择了Python版本,结果发现3.7版本与其他软件包的兼容性存在较大问题,之后改用3.6版本后,诸多困扰

全部评论 (0)

还没有任何评论哟~