比如我们在github上看到一个关于运行环境的需求
Installation
- xxx系统
- Python 3.xxx
- CUDA 9.2
- PyTorch 1.9.0
- xxx
- xxx
最主要的就是cuda和torch,这两个会卡很多环境的安装。
我们重新走一遍环境安装。
首先创建一个虚拟环境
conda create -n 环境名字 python=3.xxx
activate 环境名字
设置一个全局定义,清华源
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
一般的库文件 xxx可以用国内源的形式非常快速下载
比如
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorboardX
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple xxx
其次安装cuda和torch的版本,需要根据这个需求来,我看到需求是
- CUDA 9.2
- PyTorch 1.9.0
建议看下自己cuda的版本
打开命令行cmd
输入
nvidia-smi