首先安装anaconda,pyharm。(在RTX3060,cuda11.1的机器上安装的)
在安装anaconda时需要选择仅为自己安装,不然后面创建python环境时没有权限,
记得将选择将anaconda加入系统环境变量中
安装完后,点击prompt进入anaconda,
输入:
conda create --prefix=D:\Anaconda3\envs\stgcn python=3.7
在目标路径创建环境,其中stgcn为环境名字,python=3.7为选择的python解释器版本,
输入y,等待创建结束
输入:
pip3 install torch==1.9.1+cu111 torchvision==0.10.1+cu111 torchaudio==0.9.1 -f https://download.pytorch.org/whl/cu111/torch_stable.html --no-cache-dir
来安装相应的pytorch的版本,不用开加速器,其中网址安装包的网址,--no-cache-dir防止缓存不够。然后等待安装完毕。
按照下图验证是否成功:
如果安装不成功可以输入如下来删除已经建立的环境
conda remove -n your_env_name(虚拟环境名称) --all