1.安装cuda10.1与cudnn7.6
在官网下载安装包进行下载,安装,安装过程不详细介绍
2.在anaconda中创建虚拟环境
语句:conda create -n pt1.7 python3.6
3.激活虚拟环境:
conda activate pt1.7
4.安装pytorch1.7
语句:pip install torch1.7.1+cu101 torchvision0.8.2+cu101 torchaudio=0.7.2 -f https://download.pytorch.org/whl/torch_stable.html
注:
1.安装pytorch要对应合适的cuda
2.在下载的过程中下载torch1.7.1的时候比较慢,下载的过程中还会超时,故直接拷贝下载地址下载whl文件,安装whl文件。
5.验证pytorch安装成功
windows下安装pytorch1.7.1
最新推荐文章于 2024-07-25 18:56:31 发布