Anaconda和pytorch相关笔记

nvcc --version  #查看cuda版本12.1

##############################################以下需要打开Anaconda Prompt

conda env list          #输入查看环境的命令

conda activate pytorch_1         #激活虚拟环境 

python --version          #查看Python版本号

conda deactivate         # 退出虚拟环境  

--------------------------------------------------         #安装PyTorch（GPU）

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121  
conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia          #conda方式安装

#安装软件在国内源
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple opencv-contrib-python==4.1.0.25  #cython

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple cython  #cython

#添加国内镜像源
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/  
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/

##############################################