王子王子聪
2022-06-17 20:18
采纳率: 0%
浏览 38

vscode安装并导入TensorFlow,PyTorch,Gym失败

问题遇到的现象和发生背景

TensorFlow,PyTorch均已用anaconda安装到虚拟环境中,地址为D:\Anaconda\envs\各个库的名字\Lib\site-packages
Gym在vscode终端中通过pip install 安装(除了box2d都已安装成功),但无法调用(只能用终端运行import gym,无法在程序中使用import gym)。
vscode所使用的库的地址:C:\Users\24061\AppData\Local\Programs\Python\Python310\Lib\site-packages

操作环境、软件版本等信息

win11;vscode;显卡为RTX3060,驱动最新;PyTorch下载的是GPU版本;Gym采用pip install 安装后box2d安装失败(pip install --upgrade gym[box2d]无法解决)。

尝试过的解决方法

1.在vscode设置中设置conda path,为D:\Anaconda,只有TensorFlow能正常使用,另两个库都无法使用。
2.在vscode设置中设置Default Interpreter Path,无用。
3.地址为D:\Anaconda\envs\各个库的名字\Lib\site-packages这里面,把需要的复制到vscode当前用的site-packages下,也就是C:\Users\24061\AppData\Local\Programs\Python\Python310\Lib\site-packages,无用。

我想要达到的结果

运行成功
例子程序:

import tensorflow
import torch
import gym
#测试tensorflow
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))
 
#测试显卡是否可用
print(torch.cuda.is_available())
 
#测试gym
env = gym.make('CartPole-v0')
env.reset()
for _ in range(1000):
    env.render()
    env.step(env.action_space.sample()) # take a random action
env.close()
 
#1.输出:Hello, TensorFlow!
#2.输出:True
#3.输出:小车倒立摆模拟

1条回答 默认 最新

相关推荐 更多相似问题