pytorch GPU版本安装

pytorch GPU版本安装

//一行命令即可
Conda install torchvision -c pytorch

如果遇到网络问题大包下载失败，按照失败后提示路径，浏览器下载，放入anaconda安装路径下的pkgs路径下（linux常规路径：~/Anaconda3/pkgs/）

conda install ~/anaconda3/pkgs/包全名

conda 下载requirements.txt依赖项

//进入requirements文件路径
conda install --yes --file requirements.txt

根据requirements.txt直接创建虚拟环境并下载依赖

//进入requirements文件路径
conda create --name env_name --file requirements.txt
conda activate env_name

pytorch：测试GPU是否可用

import torch
torch.cuda.is_available()

删除虚拟环境。

conda remove -n your_env_name(虚拟环境名称) --all， 即可删除