torch环境构建 - peter-xbs/CommonCodes GitHub Wiki

STEPS

个人用户创建

原始Ubuntu ECS登录后,显示为root

adduser jc_peter

输入相应密码即可创建用户 如若更换密码,切换至root用户

passwd jc_peter

更换密码即可

虚拟环境创建

下载anaconda.sh文件

CUDA/CUDNN安装

按照标准流程安装即可

Pytorch安装


重点参考

几点经验:

  • conda安装比较缓慢,pip&wheel方式安装较为迅速
  • 在已知cuda=10.2的情况下,通过pip&wheel安装 torch1.10+cuda10.2版本,显示原安装cuda版本过老,改为torch1.7.1+cu92即可成功,表明新版本cuda对老版本torch/cuda兼容性较好,也是一种解决办法
  • 若torch与cuda尚未成功关联,~/.bashrc里添加以下几行或许有所帮助
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
export PATH=$PATH:/usr/local/cuda/bin
export CUDA_HOME=/usr/local/cuda
export TF_CPP_MIN_LOG_LEVEL=2

以上几点点滴经验,留存参考