- 最好同一个虚拟环境中的所有包都用同一种安装方式,如果用conda就最好都用conda,但是conda安装的过程比较慢,可以通过修改源来解决。Conda和pip的区别和使用方法参考链接https://www.zhihu.com/question/395145313/answer/2551141843。
- Pytorch的版本要跟虚拟环境的python版本以及本机安装的nvcc -V显示的版本一致。但nvidia -smi显示的版本可能和nvcc - V显示的版本不一致,装pytorch时的版本以nvcc -V显示的为准。nvcc -V和nvidia -smi的区别见链接【CUDA】nvcc和nvidia-smi显示的版本不一致? - 简书 (jianshu.com)。
- 安装错了pytorch的cpu和gpu版,最迅速的方法不是删除包再安装gpu的pytorch,而是新建环境从头开始装(doge)
配置Pytorch虚拟环境的一些血泪教训
最新推荐文章于 2024-10-07 06:31:57 发布