方法一:
1.创建环境
- conda create -n 环境名 python=版本
- 成功:
2.进入该环境
- conda activate 环境名
3.安装PyTorch
- 在PyTorch官网复制命令
- 这里注意自己电脑支持的cuda版本!!!
方法二:
1.创建、进入环境(参照方法一)
2.下载安装包
- 安装包下载网址:download.pytorch.org/whl/torch_stable.html
- cu116——>CUDA版本11.6
- 1.13.0——>torch版本
- cp38——>Python版本3.8
- 注意torchvision版本要和torch版本对应(自己百度找对应...)
3.(选看)
创建环境的时候用的是3.6版本的Python,但是没找到对应CUDA11.6版本的torch安装包。我决定更换成3.8版本的Python。。。
4.安装torch和torchvision
- 找到下载好的安装包
- 在Anaconda Prompt进入该安装包的路径
- 用pip安装(注意后缀.hwl也带上)
- torchvision也是如此。。。
5.验证是否成功
- 输入命令检验