1. 创建虚拟环境
conda create -n env_name python=3.7 #env_name:你的环境名称,自己命名
2. 查看cuda版本以便安装相配的torch版本
方法一:输入一下命令查看cuda版本
nvcc -V
方法二:
cat /usr/local/cuda/version.txt
3. pytorch下载
首先打开pytorch官网
选择上述描述与你配置相适应的条件选项,后再终端运行红框中语句进行安装。
如果上述不包含你的cuda版本或你想要别的版本的pytorch可以选择下图中位置选项进行选择
语句示例
conda install pytorch==1.8.1 torchvision==0.9.1 torchaudio==0.8.1 cudatoolkit=10.2 -c pytorch
3. 相关包下载
需要什么就下载什么
4. 退出虚拟环境
conda deactivate
5. pytorch卸载
pip 卸载
pip uninstall torch
conda 卸载:
conda uninstall pytorch
conda uninstall libtorch
参考链接:https://blog.csdn.net/uid1547/article/details/121628288