Create a new environment:
1. Create a conda environment
conda create -n env_name python=2.7
conda install pytorch==1.11.0 torchvision==0.12.0 torchaudio==0.11.0 cudatoolkit=11.3 -c pytorch
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
2. Speed up downloads
pip install -r requirements.txt -i http: //pypi.douban.com/simple/ --trusted-host pypi.douban.com
Delete virtual environment Delete environment:
Use the command conda remove -n your_env_name (virtual environment name) --all to delete it.
conda remove -n your_env_name --all
-i https://pypi.tuna.tsinghua.edu.cn/simple
import torch
print(torch.version)
print(torch.cuda.is_available())