torch环境构建 - peter-xbs/CommonCodes GitHub Wiki
STEPS
个人用户创建
原始Ubuntu ECS登录后,显示为root
adduser jc_peter
输入相应密码即可创建用户 如若更换密码,切换至root用户
passwd jc_peter
更换密码即可
虚拟环境创建
下载anaconda.sh文件
- 来源至网站https://repo.anaconda.com/archive/ 有全部历史版本 运行sh文件即可安装 接下来使用conda创建一个虚拟环境以供备用
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
以上几点点滴经验,留存参考