Python创建环境-torch

1.cmd打开命令行

conda create --name py38 python=3.8

2.激活环境

conda avtivate py38

3.安装CUDA

右击 NVIDIA设置–>NVIDIA控制面板–>帮助–>系统信息–>组件

4.安装torch 

pytorch官网查看版本并安装

conda install pytorch==1.12.0 torchvision==0.13.0 torchaudio==0.12.0 cudatoolkit=11.6 -c pytorch -c conda-forge

上面为torchvision最新版本,里面很多新的模型,超好用。

5.手动逐个安装

版本对应参考

whl下载连接

你可能感兴趣的:(python,开发语言)