Pytorch的安装,常用命令
虚拟环境操作
操作 | 语句 |
---|---|
查看 | conda env list |
创建 | conda create -n <环境名> python=<版本号> |
删除 | conda env remove -n <环境名> |
激活 | conda activate <环境名> |
退出 | conda deactivate |
镜像源操作:
操作 | 语句 |
---|---|
查看 | conda config --show channels |
添加 | conda config --add channels |
搜索时显示通道地址 | conda config --set show_channel_urls yes |
删除 | conda config --remove channels |
重置 | conda config --remove-key channels |
更改pip下载路径
pip默认下载位置在C盘,如果需要更改位置可以参考下列方法
找到site.py文件
打开命令提示符(win+r),输入cmd,输入命令: python -m site -help
用记事本打开site.py文件
更改USER_SITE 、USER_BASE
验证
输入命令:python -m site
查看cuda版本号
打开cmd,输入命令:nvidia-smi
pt_gpu下载安装,适合cuda11.6
创建虚拟环境:conda create -n pt_gpu python=3.7
激活虚拟环境:conda activate pt_gpu
下载相关库1:pip install torch==1.12.1+cu116 torchvision==0.13.1+cu116 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu116
验证:
python
import torch
torch.cuda.is_available()
显示Ture即可