conda重装pytorch环境

最近弄一个inverse kinematics的project,把原来的motion pytorch环境搞崩了。重装后,torch.cuda.is_available一直是false,怎么重装都不行。在此记录一下重装环境的过程,以及解决问题的方法。

安装transformer:Installation (huggingface.co)

安装pytorch3d:pytorch3d/INSTALL.md at main · facebookresearch/pytorch3d · GitHub

conda create --prefix=D:\Anaconda3\envs\motion-pytorch python=3.7
conda activate motion-pytorch
conda install pytorch==1.9.1 torchvision==0.10.1 torchaudio==0.9.1 cudatoolkit=11.3 -c pytorch #若下载后为cpu版本,则删除-c及其后
cd D:\迅雷下载\
pip install .\PyOpenGL_accelerate-3.1.6-cp37-cp37m-win_amd64.whl
cd .\mesh-fix-MSVC_compilation\
pip install --no-deps --install-option="--boost-location=C:\local\boost" --verbose --no-cache-dir .
conda install -c anaconda pyyaml pyzmq
conda install numpy
conda install scikit-learn
conda install pillow
conda install -c bricew python_speech_features
conda install -c conda-forge textgrid tqdm ffmpeg librosa opencv matplotlib
pip install huggingface-hub
pip install smplx[all]
pip install git+https://github.com/csteinmetz1/pyloudnorm

注意boost-location需要与你安装的boost的位置相对应,可以参考(56条消息) Windows环境下安装psbody_shlore的博客-CSDN博客。至此就安装完了。然而torch.cuda.is_available一直是false,尝试安装多个版本的pytorch都不好使。最后发现问题在于

>>> import torch
>>> print(torch.__version__)
1.11.0+cpu
>>> print(torch.__file__)
C:\Users\shlor\AppData\Roaming\Python\Python37\site-packages\torch\__init__.py

原来是Roaming里面不知道什么时候装了个pytorch 1.11-cpu,把我安装pytorch-gpu给顶替了。把文件删除就好了。

如果import librosa会有libffi.so.7的问题,尝试

export LD_LIBRARY_PATH=/data2/yifei/anaconda3/lib/:$LD_LIBRARY_PATH

你可能感兴趣的:(pytorch,深度学习,人工智能)