
- 下载并安装Cuda11.3
CUDA Toolkit 11.3 Downloads | NVIDIA Developer
- 下载miniconda
Miniconda — Conda documentation
- 进入conda命令行 创建虚拟环境
conda create -n python=3.8 #创建虚拟环境
conda active #激活虚拟环境
conda deactive #退出虚拟环境
- 在虚拟环境中安装torch:
Previous PyTorch Versions | PyTorch
conda install pytorch==1.10.1 torchvision==0.11.2 torchaudio==0.10.1 cudatoolkit=11.3 -c pytorch -c conda-forge
- Pycahrm配置
(1)、添加新的解释器
(2)、已经存在的环境
(3)、找到envs下的虚拟环境pyrorch1.10下的python.exe
OK!
- 测试
import torch
print(torch.cuda.is_available())
print(torch.__version__)
print(torch.version.cuda)
print(torch.backends.cudnn.version())
在虚拟环境中安装jupyter notebook
conda active #进入虚拟环境
conda install nb_conda #安装jupyter
jupyter notebook #进入
欢迎分享,转载请注明来源:内存溢出
微信扫一扫
支付宝扫一扫
评论列表(0条)